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

Class ClusterLearner

java.lang.Object

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

Direct Known Subclasses:

ClusterLearnerAvg

public class ClusterLearner
extends Object

Constructor Summary

Constructors

Constructor and Description
ClusterLearner(int numEpochs, Set trainingDocuments, Pipe p, int yesIndex, int noIndex)
ClusterLearner(int numEpochs, Set trainingDocuments, Pipe p, MaxEnt classifier, int yesIndex, int noIndex)

Method Summary

Modifier and Type	Method and Description
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(HashMap map)
void	startTraining(Set testDocInstances)
protected void	testCurrentModel(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,
                      Set trainingDocuments,
                      Pipe p,
                      MaxEnt classifier,
                      int yesIndex,
                      int noIndex)

ClusterLearner

public ClusterLearner(int numEpochs,
                      Set trainingDocuments,
                      Pipe p,
                      int yesIndex,
                      int noIndex)

Method Detail

getInitializedMatrix

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

initializePrevClusterings

public void initializePrevClusterings(HashMap map)

startTraining

public void startTraining(Set testDocInstances)

testCurrentModel

protected void testCurrentModel(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)