cc.mallet.topics

Class MarginalProbEstimator

java.lang.Object

cc.mallet.topics.MarginalProbEstimator

All Implemented Interfaces:

java.io.Serializable

public class MarginalProbEstimator
extends java.lang.Object
implements java.io.Serializable

An implementation of topic model marginal probability estimators presented in Wallach et al., "Evaluation Methods for Topic Models", ICML (2009)

See Also:

Serialized Form

Constructor Summary

Constructors

Constructor and Description
MarginalProbEstimator(int numTopics, double[] alpha, double alphaSum, double beta, int[][] typeTopicCounts, int[] tokensPerTopic)

Method Summary

Modifier and Type	Method and Description
double	evaluateLeftToRight(InstanceList testing, int numParticles, boolean usingResampling, java.io.PrintStream docProbabilityStream)
int[]	getTokensPerTopic()
int[][]	getTypeTopicCounts()
static MarginalProbEstimator	read(java.io.File f)
void	setPrintWords(boolean shouldPrint)
void	setRandom(Randoms r)
void	setRandomSeed(int seed)

Methods inherited from class java.lang.Object

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

Constructor Detail

MarginalProbEstimator

public MarginalProbEstimator(int numTopics,
                             double[] alpha,
                             double alphaSum,
                             double beta,
                             int[][] typeTopicCounts,
                             int[] tokensPerTopic)

Method Detail

setRandomSeed

public void setRandomSeed(int seed)

setRandom

public void setRandom(Randoms r)

getTokensPerTopic

public int[] getTokensPerTopic()

getTypeTopicCounts

public int[][] getTypeTopicCounts()

setPrintWords

public void setPrintWords(boolean shouldPrint)

evaluateLeftToRight

public double evaluateLeftToRight(InstanceList testing,
                                  int numParticles,
                                  boolean usingResampling,
                                  java.io.PrintStream docProbabilityStream)

read

public static MarginalProbEstimator read(java.io.File f)
                                  throws java.lang.Exception

Throws:

java.lang.Exception