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 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)
|
Methods inherited from class java.lang.Object |
clone, equals, finalize, getClass, hashCode, notify, notifyAll, toString, wait, wait, wait |
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)
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)