edu.umass.cs.mallet.projects.seg_plus_coref.coreference

Class CorefClusterAdv

java.lang.Object

edu.umass.cs.mallet.projects.seg_plus_coref.coreference.CorefClusterAdv

Direct Known Subclasses:

MultipleCorefClusterer

public class CorefClusterAdv
extends Object

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
class	CorefClusterAdv.EdgeComparator
class	CorefClusterAdv.PseudoEdge
class	CorefClusterAdv.PseudoEdgeComparator
class	CorefClusterAdv.PseudoVertex

Constructor Summary

Constructors

Constructor and Description
CorefClusterAdv()
CorefClusterAdv(double threshold)
CorefClusterAdv(double threshold, Matrix2 sgdParameters, int numSGDFeatures)
CorefClusterAdv(double threshold, MaxEnt classifier, Pipe p)
CorefClusterAdv(double threshold, MaxEnt classifier, TreeModel tm, Pipe p)
CorefClusterAdv(Pipe p)
CorefClusterAdv(Pipe p, TreeModel tm)
CorefClusterAdv(TreeModel tm)

Method Summary

Modifier and Type	Method and Description
Collection	absoluteCluster(InstanceList ilist, List mentions)
void	addVerticesToGraph(salvo.jesus.graph.WeightedGraph graph, List mentions, HashMap alreadyAddedVertices)
protected salvo.jesus.graph.WeightedGraph	buildGraphFromPseudoEdges(List pedges, List mentions)
salvo.jesus.graph.WeightedEdge	chooseEdge2(List edges, double minVal, double total, Random rand)
salvo.jesus.graph.WeightedEdge	chooseEdge3(List edges, double minVal, double total, Random rand)
Collection	clusterMentions(InstanceList ilist, List mentions)
Collection	clusterMentions(InstanceList ilist, List mentions, int optimalNBest, boolean stochastic)
protected double	collectionAvg(Collection collection)
void	completeGraphNBest(salvo.jesus.graph.WeightedGraph graph, Collection keyPartitioning, Map citMap)
double	computeInitialTreeObjScore(Collection pvertices)
protected void	constructEdgesFromPseudoEdges(salvo.jesus.graph.WeightedGraph graph, CorefClusterAdv.PseudoEdge pedge, HashMap alreadyAdded)
void	constructEdgesUsingTrainedClusterer(salvo.jesus.graph.WeightedGraph graph, Instance instPair, HashMap alreadyAdded)
void	constructEdgesUsingTrainedClusterer(salvo.jesus.graph.WeightedGraph graph, Instance instPair, HashMap alreadyAdded, Double edgeWeight)
void	constructEdgesUsingTrainedClusterer(salvo.jesus.graph.WeightedGraph graph, Instance instPair, HashMap alreadyAdded, Double edgeWeight, MaxEnt classifier)
salvo.jesus.graph.WeightedGraph	constructOptimalEdgesUsingNBest(List mentions, int N)
salvo.jesus.graph.WeightedGraph	copyGraph(salvo.jesus.graph.WeightedGraph graph)
salvo.jesus.graph.WeightedGraph	createGraph(InstanceList ilist, List mentions)
salvo.jesus.graph.WeightedGraph	createGraph(InstanceList ilist, List mentions, salvo.jesus.graph.WeightedGraph graph)
salvo.jesus.graph.WeightedGraph	createGraph(InstanceList ilist, List mentions, salvo.jesus.graph.WeightedGraph graph, MaxEnt classifier)
List	createPseudoEdges(InstanceList instances, Map map)
Collection	createPseudoVertices(InstanceList instances, List mentions, HashMap map)
double	evaluateAgainstKey(Collection col)
double	evaluatePartitioning(Collection clustering, salvo.jesus.graph.WeightedGraph graph)
double	evaluatePartitioningExternal(InstanceList ilist, List mentions, Collection collection)
double	evaluatePartitioningExternal(InstanceList ilist, List mentions, Collection collection, int nBestList)
void	exportGraph(String file)
MaxEnt	getClassifier()
protected Collection	getClusteringFromPseudo(Collection pvertices)
Collection	getCollectionOfOriginalObjects(Collection vertices) Construct a Collection of Collections where the objects in the collections are the original objects (i.e.
protected Collection	getPseudoClustering(Collection pvertices)
void	getUnNormalizedScores(Matrix2 lambdas, FeatureVector fv, double[] scores)
protected boolean	hasNextIndexList(int[] indexList, int N)
boolean	inSameCluster(Collection clustering, Object o1, Object o2)
void	loadME(String file)
void	mergeVertices(salvo.jesus.graph.WeightedGraph g, salvo.jesus.graph.VertexImpl v1, salvo.jesus.graph.VertexImpl v2)
protected int[]	nextIndexList(int[] indexList, int N)
protected int[]	nextIndexListStochastic(int[] indexList, int N)
protected int	numSingletons(Collection clustering)
Collection	partitionGraph(salvo.jesus.graph.WeightedGraph origGraph)
void	printGraph(salvo.jesus.graph.WeightedGraph g)
String	printParamDetails(FeatureVector vec, Classification classification, MaxEnt classifier)
void	printParams(MaxEnt me)
void	setConfWeightedScores(boolean b)
void	setFullPartition(boolean f)
void	setKeyPartitioning(Collection keyP)
void	setNBestInference(boolean b)
void	setOptimality(boolean b)
void	setRBeamSize(int s)
void	setSearchParams(int iters, int reductions)
void	setThreshold(double t)
void	setTrueNumStop(boolean b)
void	testClassifier(InstanceList tlist)
void	testClassifier(InstanceList tlist, MaxEnt classifier)
void	train(InstanceList ilist)
MaxEnt	trainClassifier(InstanceList ilist)
Collection	typicalClusterAdv(InstanceList ilist, List mentions)
Collection	typicalClusterPartition(salvo.jesus.graph.WeightedGraph graph)
void	updateGraphNBest(salvo.jesus.graph.WeightedGraph graph, int[] indexList, List instList, HashMap alreadyAdded)
double	updateScore(double curScore, double[] treeScore, CorefClusterAdv.PseudoVertex v1, CorefClusterAdv.PseudoVertex v2, Set s1, Set s2, boolean over_ride)
double	weightOfConfig(int[] indexList, List instList)

Methods inherited from class java.lang.Object

clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait

Constructor Detail

CorefClusterAdv

public CorefClusterAdv()

CorefClusterAdv

public CorefClusterAdv(TreeModel tm)

CorefClusterAdv

public CorefClusterAdv(Pipe p)

CorefClusterAdv

public CorefClusterAdv(Pipe p,
                       TreeModel tm)

CorefClusterAdv

public CorefClusterAdv(double threshold)

CorefClusterAdv

public CorefClusterAdv(double threshold,
                       MaxEnt classifier,
                       Pipe p)

CorefClusterAdv

public CorefClusterAdv(double threshold,
                       MaxEnt classifier,
                       TreeModel tm,
                       Pipe p)

CorefClusterAdv

public CorefClusterAdv(double threshold,
                       Matrix2 sgdParameters,
                       int numSGDFeatures)

Method Detail

setConfWeightedScores

public void setConfWeightedScores(boolean b)

setRBeamSize

public void setRBeamSize(int s)

setOptimality

public void setOptimality(boolean b)

setNBestInference

public void setNBestInference(boolean b)

setTrueNumStop

public void setTrueNumStop(boolean b)

setSearchParams

public void setSearchParams(int iters,
                            int reductions)

setThreshold

public void setThreshold(double t)

setKeyPartitioning

public void setKeyPartitioning(Collection keyP)

setFullPartition

public void setFullPartition(boolean f)

loadME

public void loadME(String file)

train

public void train(InstanceList ilist)

trainClassifier

public MaxEnt trainClassifier(InstanceList ilist)

testClassifier

public void testClassifier(InstanceList tlist)

testClassifier

public void testClassifier(InstanceList tlist,
                           MaxEnt classifier)

printParamDetails

public String printParamDetails(FeatureVector vec,
                                Classification classification,
                                MaxEnt classifier)

printParams

public void printParams(MaxEnt me)

getClassifier

public MaxEnt getClassifier()

clusterMentions

public Collection clusterMentions(InstanceList ilist,
                                  List mentions)

clusterMentions

public Collection clusterMentions(InstanceList ilist,
                                  List mentions,
                                  int optimalNBest,
                                  boolean stochastic)

createGraph

public salvo.jesus.graph.WeightedGraph createGraph(InstanceList ilist,
                                                   List mentions)

createGraph

public salvo.jesus.graph.WeightedGraph createGraph(InstanceList ilist,
                                                   List mentions,
                                                   salvo.jesus.graph.WeightedGraph graph)

createGraph

public salvo.jesus.graph.WeightedGraph createGraph(InstanceList ilist,
                                                   List mentions,
                                                   salvo.jesus.graph.WeightedGraph graph,
                                                   MaxEnt classifier)

exportGraph

public void exportGraph(String file)

copyGraph

public salvo.jesus.graph.WeightedGraph copyGraph(salvo.jesus.graph.WeightedGraph graph)

addVerticesToGraph

public void addVerticesToGraph(salvo.jesus.graph.WeightedGraph graph,
                               List mentions,
                               HashMap alreadyAddedVertices)

chooseEdge3

public salvo.jesus.graph.WeightedEdge chooseEdge3(List edges,
                                                  double minVal,
                                                  double total,
                                                  Random rand)

chooseEdge2

public salvo.jesus.graph.WeightedEdge chooseEdge2(List edges,
                                                  double minVal,
                                                  double total,
                                                  Random rand)

evaluatePartitioningExternal

public double evaluatePartitioningExternal(InstanceList ilist,
                                           List mentions,
                                           Collection collection)

evaluatePartitioningExternal

public double evaluatePartitioningExternal(InstanceList ilist,
                                           List mentions,
                                           Collection collection,
                                           int nBestList)

evaluatePartitioning

public double evaluatePartitioning(Collection clustering,
                                   salvo.jesus.graph.WeightedGraph graph)

inSameCluster

public boolean inSameCluster(Collection clustering,
                             Object o1,
                             Object o2)

createPseudoEdges

public List createPseudoEdges(InstanceList instances,
                              Map map)

createPseudoVertices

public Collection createPseudoVertices(InstanceList instances,
                                       List mentions,
                                       HashMap map)

updateScore

public double updateScore(double curScore,
                          double[] treeScore,
                          CorefClusterAdv.PseudoVertex v1,
                          CorefClusterAdv.PseudoVertex v2,
                          Set s1,
                          Set s2,
                          boolean over_ride)

computeInitialTreeObjScore

public double computeInitialTreeObjScore(Collection pvertices)

absoluteCluster

public Collection absoluteCluster(InstanceList ilist,
                                  List mentions)

getPseudoClustering

protected Collection getPseudoClustering(Collection pvertices)

typicalClusterAdv

public Collection typicalClusterAdv(InstanceList ilist,
                                    List mentions)

numSingletons

protected int numSingletons(Collection clustering)

getClusteringFromPseudo

protected Collection getClusteringFromPseudo(Collection pvertices)

buildGraphFromPseudoEdges

protected salvo.jesus.graph.WeightedGraph buildGraphFromPseudoEdges(List pedges,
                                                                    List mentions)

typicalClusterPartition

public Collection typicalClusterPartition(salvo.jesus.graph.WeightedGraph graph)

partitionGraph

public Collection partitionGraph(salvo.jesus.graph.WeightedGraph origGraph)

evaluateAgainstKey

public double evaluateAgainstKey(Collection col)

getCollectionOfOriginalObjects

public Collection getCollectionOfOriginalObjects(Collection vertices)

Construct a Collection of Collections where the objects in the collections are the original objects (i.e. the object of the vertices)

mergeVertices

public void mergeVertices(salvo.jesus.graph.WeightedGraph g,
                          salvo.jesus.graph.VertexImpl v1,
                          salvo.jesus.graph.VertexImpl v2)

printGraph

public void printGraph(salvo.jesus.graph.WeightedGraph g)

collectionAvg

protected double collectionAvg(Collection collection)

hasNextIndexList

protected boolean hasNextIndexList(int[] indexList,
                                   int N)

nextIndexList

protected int[] nextIndexList(int[] indexList,
                              int N)

nextIndexListStochastic

protected int[] nextIndexListStochastic(int[] indexList,
                                        int N)

weightOfConfig

public double weightOfConfig(int[] indexList,
                             List instList)

updateGraphNBest

public void updateGraphNBest(salvo.jesus.graph.WeightedGraph graph,
                             int[] indexList,
                             List instList,
                             HashMap alreadyAdded)

constructOptimalEdgesUsingNBest

public salvo.jesus.graph.WeightedGraph constructOptimalEdgesUsingNBest(List mentions,
                                                                       int N)

completeGraphNBest

public void completeGraphNBest(salvo.jesus.graph.WeightedGraph graph,
                               Collection keyPartitioning,
                               Map citMap)

constructEdgesFromPseudoEdges

protected void constructEdgesFromPseudoEdges(salvo.jesus.graph.WeightedGraph graph,
                                             CorefClusterAdv.PseudoEdge pedge,
                                             HashMap alreadyAdded)

constructEdgesUsingTrainedClusterer

public void constructEdgesUsingTrainedClusterer(salvo.jesus.graph.WeightedGraph graph,
                                                Instance instPair,
                                                HashMap alreadyAdded)

constructEdgesUsingTrainedClusterer

public void constructEdgesUsingTrainedClusterer(salvo.jesus.graph.WeightedGraph graph,
                                                Instance instPair,
                                                HashMap alreadyAdded,
                                                Double edgeWeight)

constructEdgesUsingTrainedClusterer

public void constructEdgesUsingTrainedClusterer(salvo.jesus.graph.WeightedGraph graph,
                                                Instance instPair,
                                                HashMap alreadyAdded,
                                                Double edgeWeight,
                                                MaxEnt classifier)

getUnNormalizedScores

public void getUnNormalizedScores(Matrix2 lambdas,
                                  FeatureVector fv,
                                  double[] scores)