edu.umass.cs.mallet.projects.seg_plus_coref.clustering
Class ClusterLearner

java.lang.Object
  extended byedu.umass.cs.mallet.projects.seg_plus_coref.clustering.ClusterLearner
Direct Known Subclasses:
ClusterLearnerAvg

public class ClusterLearner
extends java.lang.Object


Constructor Summary
ClusterLearner(int numEpochs, java.util.Set trainingDocuments, Pipe p, int yesIndex, int noIndex)
           
ClusterLearner(int numEpochs, java.util.Set trainingDocuments, Pipe p, MaxEnt classifier, int yesIndex, int noIndex)
           
 
Method Summary
protected  void constructEdges(MappedGraph graph, Instance pair, Matrix2 lambdas)
           
protected  DenseVector getDenseVectorOf(int ri, Matrix2 matrix)
           
 Matrix2 getFinalLambdas()
           
protected  double[][] getInitializedMatrix(int d1, int d2)
           
 void getUnNormalizedScores(Matrix2 lambdas, FeatureVector fv, double[] scores)
           
 void initializePrevClusterings(java.util.HashMap map)
           
 void startTraining(java.util.Set testDocInstances)
           
protected  void testCurrentModel(java.util.Set testDocInstances, Matrix2 lambdas)
           
 
Methods inherited from class java.lang.Object
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait
 

Constructor Detail

ClusterLearner

public ClusterLearner(int numEpochs,
                      java.util.Set trainingDocuments,
                      Pipe p,
                      MaxEnt classifier,
                      int yesIndex,
                      int noIndex)

ClusterLearner

public ClusterLearner(int numEpochs,
                      java.util.Set trainingDocuments,
                      Pipe p,
                      int yesIndex,
                      int noIndex)
Method Detail

getInitializedMatrix

protected double[][] getInitializedMatrix(int d1,
                                          int d2)

initializePrevClusterings

public void initializePrevClusterings(java.util.HashMap map)

startTraining

public void startTraining(java.util.Set testDocInstances)

testCurrentModel

protected void testCurrentModel(java.util.Set testDocInstances,
                                Matrix2 lambdas)

getDenseVectorOf

protected DenseVector getDenseVectorOf(int ri,
                                       Matrix2 matrix)

getFinalLambdas

public Matrix2 getFinalLambdas()

getUnNormalizedScores

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

constructEdges

protected void constructEdges(MappedGraph graph,
                              Instance pair,
                              Matrix2 lambdas)