Package org.dice_research.topicmodeling.algorithm.mallet

Class MalletLdaWrapper

java.lang.Object

org.dice_research.topicmodeling.algorithm.mallet.MalletLdaWrapper

All Implemented Interfaces:

Serializable, org.dice_research.topicmodeling.algorithms.ModelingAlgorithm, org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier, org.dice_research.topicmodeling.algorithms.VocabularyContaining, org.dice_research.topicmodeling.preprocessing.PreprocessorFactory

public class MalletLdaWrapper
extends Object
implements org.dice_research.topicmodeling.algorithms.ModelingAlgorithm, org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier

See Also:

Serialized Form

Nested Class Summary

Nested Classes

Modifier and Type	Class	Description
protected static class	MalletLdaWrapper.MalletLDATopicModeler

Field Summary

Fields

Modifier and Type	Field	Description
protected MalletAlphabetWrapper	alphabet
private static org.slf4j.Logger	LOGGER
private static String	MALLET_REGEX_TOKEN
private String	malletRegexToken
protected long	seed
private static long	serialVersionUID
protected MalletLdaWrapper.MalletLDATopicModeler	topicModel
protected org.dice_research.topicmodeling.algorithms.WordCounter	wordCounter

Constructor Summary

Constructors

Constructor	Description
MalletLdaWrapper(int numberOfTopics)
MalletLdaWrapper(int numberOfTopics, double alphaSum, double beta)
MalletLdaWrapper(int numberOfTopics, double alphaSum, double beta, long seed)
MalletLdaWrapper(int numberOfTopics, long seed)
MalletLdaWrapper(cc.mallet.types.LabelAlphabet topicAlphabet, double alphaSum, double beta)
MalletLdaWrapper(cc.mallet.types.LabelAlphabet topicAlphabet, double alphaSum, double beta, long seed)

Method Summary

Modifier and Type	Method	Description
protected cc.mallet.types.Instance	createInstanceFromDocument(org.dice_research.topicmodeling.utils.doc.Document document, cc.mallet.types.Alphabet alphabet)
protected void	createMultipleSpellingVocabulary(org.dice_research.topicmodeling.utils.corpus.DocumentListCorpus<?> corpus)
org.dice_research.topicmodeling.preprocessing.Preprocessor	createPreprocessor(org.dice_research.topicmodeling.preprocessing.docsupplier.DocumentSupplier supplier, org.dice_research.topicmodeling.lang.Language lang)
private void	directInitialization(org.dice_research.topicmodeling.utils.corpus.Corpus corpus, org.dice_research.topicmodeling.utils.vocabulary.Vocabulary vocabulary)
cc.mallet.types.FeatureSequence	getDocumentAsFeatureSequence(int documentId)
org.dice_research.topicmodeling.algorithms.Model	getModel()
int	getNumberOfDocuments()
int	getNumberOfTopics()
int	getNumberOfWords()
long	getSeed()
protected String[]	getTokenizedTermsAsText(org.dice_research.topicmodeling.utils.corpus.DocumentListCorpus<?> corpus)
org.dice_research.topicmodeling.utils.vocabulary.Vocabulary	getVocabulary()
org.dice_research.topicmodeling.algorithms.WordCounter	getWordCounts()
int[]	getWordsOfDocument(int documentId)
int[]	getWordTopicAssignmentForDocument(int documentId)
void	initialize(org.dice_research.topicmodeling.utils.corpus.Corpus corpus)
void	performNextStep()
void	setMalletRegexToken(String malletRegexToken)
void	setOptimizeInterval(int interval)	Set to 0 to turn optimization off.

Methods inherited from class java.lang.Object

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

Field Details

serialVersionUID

private static final long serialVersionUID

See Also:

Constant Field Values

LOGGER

private static final org.slf4j.Logger LOGGER

MALLET_REGEX_TOKEN

private static final String MALLET_REGEX_TOKEN

See Also:

Constant Field Values

topicModel

protected MalletLdaWrapper.MalletLDATopicModeler topicModel

alphabet

protected MalletAlphabetWrapper alphabet

seed

protected long seed

wordCounter

protected transient org.dice_research.topicmodeling.algorithms.WordCounter wordCounter

malletRegexToken

private String malletRegexToken

Constructor Details

MalletLdaWrapper

public MalletLdaWrapper(int numberOfTopics)

MalletLdaWrapper

public MalletLdaWrapper(int numberOfTopics,
 long seed)

MalletLdaWrapper

public MalletLdaWrapper(int numberOfTopics,
 double alphaSum,
 double beta)

MalletLdaWrapper

public MalletLdaWrapper(int numberOfTopics,
 double alphaSum,
 double beta,
 long seed)

MalletLdaWrapper

public MalletLdaWrapper(cc.mallet.types.LabelAlphabet topicAlphabet,
 double alphaSum,
 double beta)

MalletLdaWrapper

public MalletLdaWrapper(cc.mallet.types.LabelAlphabet topicAlphabet,
 double alphaSum,
 double beta,
 long seed)

Method Details

createPreprocessor

public org.dice_research.topicmodeling.preprocessing.Preprocessor createPreprocessor(org.dice_research.topicmodeling.preprocessing.docsupplier.DocumentSupplier supplier,
 org.dice_research.topicmodeling.lang.Language lang)

Specified by:

createPreprocessor in interface org.dice_research.topicmodeling.preprocessing.PreprocessorFactory

initialize

public void initialize(org.dice_research.topicmodeling.utils.corpus.Corpus corpus)

Specified by:

initialize in interface org.dice_research.topicmodeling.algorithms.ModelingAlgorithm

getTokenizedTermsAsText

protected String[] getTokenizedTermsAsText(org.dice_research.topicmodeling.utils.corpus.DocumentListCorpus<?> corpus)

directInitialization

private void directInitialization(org.dice_research.topicmodeling.utils.corpus.Corpus corpus,
 org.dice_research.topicmodeling.utils.vocabulary.Vocabulary vocabulary)

createInstanceFromDocument

protected cc.mallet.types.Instance createInstanceFromDocument(org.dice_research.topicmodeling.utils.doc.Document document,
 cc.mallet.types.Alphabet alphabet)

createMultipleSpellingVocabulary

protected void createMultipleSpellingVocabulary(org.dice_research.topicmodeling.utils.corpus.DocumentListCorpus<?> corpus)

performNextStep

public void performNextStep()

Specified by:

performNextStep in interface org.dice_research.topicmodeling.algorithms.ModelingAlgorithm

getModel

public org.dice_research.topicmodeling.algorithms.Model getModel()

Specified by:

getModel in interface org.dice_research.topicmodeling.algorithms.ModelingAlgorithm

getWordTopicAssignmentForDocument

public int[] getWordTopicAssignmentForDocument(int documentId)

Specified by:

getWordTopicAssignmentForDocument in interface org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier

getWordsOfDocument

public int[] getWordsOfDocument(int documentId)

Specified by:

getWordsOfDocument in interface org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier

getDocumentAsFeatureSequence

public cc.mallet.types.FeatureSequence getDocumentAsFeatureSequence(int documentId)

getNumberOfTopics

public int getNumberOfTopics()

Specified by:

getNumberOfTopics in interface org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier

getNumberOfDocuments

public int getNumberOfDocuments()

Specified by:

getNumberOfDocuments in interface org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier

getNumberOfWords

public int getNumberOfWords()

Specified by:

getNumberOfWords in interface org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier

getVocabulary

public org.dice_research.topicmodeling.utils.vocabulary.Vocabulary getVocabulary()

Specified by:

getVocabulary in interface org.dice_research.topicmodeling.algorithms.VocabularyContaining

getSeed

public long getSeed()

Specified by:

getSeed in interface org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier

getWordCounts

public org.dice_research.topicmodeling.algorithms.WordCounter getWordCounts()

Specified by:

getWordCounts in interface org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier

setMalletRegexToken

public void setMalletRegexToken(String malletRegexToken)

setOptimizeInterval

public void setOptimizeInterval(int interval)

Set to 0 to turn optimization off.

Parameters:

interval -