org.dice_research.topicmodeling.io

Class ModelCSVWriter

java.lang.Object

org.dice_research.topicmodeling.io.AbstractModelFilesWriter

org.dice_research.topicmodeling.io.ModelCSVWriter

All Implemented Interfaces:

CSVFileProcessor, ModelWriter

public class ModelCSVWriter
extends AbstractModelFilesWriter
implements CSVFileProcessor

Nested Class Summary

Nested Classes

Modifier and Type	Class and Description
private static class	ModelCSVWriter.TopicProbabilityForDocument
protected static class	ModelCSVWriter.TopWordContainer

Field Summary

Fields

Modifier and Type	Field and Description
protected String	fileNamePrefix
protected NumberFormat	formatter
private static org.slf4j.Logger	logger
protected int	numberOfTopTopicsPerDocument
protected int	numberOfTopWordsPerTopic
protected boolean	printWordTopicProbs
protected org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier	probTopicStateSupplier
private static String	SEPARATING_TOP_WORDS_FILE
private static String	TOP_WORDS_FILE
private static String[]	TOPIC_PROBS_FILE_HEAD
private static String	TOPIC_PROBS_FILE_NAME
private static String	TOPICS_FOR_DOCUMENTS_FILE
private static String	WORD_TOPIC_PROBS_FILE_NAME
private static String	WORD_TOPIC_SEPARATING_PROBS_FILE_NAME

Fields inherited from class org.dice_research.topicmodeling.io.AbstractModelFilesWriter

folder

Fields inherited from interface org.dice_research.topicmodeling.io.CSVFileProcessor

ESCAPECHAR, QUOTECHAR, SEPARATOR

Constructor Summary

Constructors

Constructor and Description
ModelCSVWriter(File folder, String fileNamePrefix)
ModelCSVWriter(org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier probTopicStateSupplier, File folder)
ModelCSVWriter(org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier probTopicStateSupplier, File folder, boolean printWordTopicProbs)

Method Summary

Modifier and Type	Method and Description
protected ModelCSVWriter.TopWordContainer	processSeparatingWordTopicProbabilities(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)
protected ModelCSVWriter.TopWordContainer	processWordTopicProbabilities(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)
void	setNumberOfTopWordsPerTopic(int numberOfTopWordsPerTopic)
protected void	writeModelToCSVFile(org.dice_research.topicmodeling.algorithms.Model model)
void	writeModelToFiles(org.dice_research.topicmodeling.algorithms.Model model)
protected void	writeProbabilisticWordTopicModel(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)
protected void	writeSeparatingTopWords(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model, ModelCSVWriter.TopWordContainer topWords)
protected void	writeTopicFile(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)
protected void	writeTopicsForDocuments(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)
protected void	writeTopWords(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model, ModelCSVWriter.TopWordContainer topWords)

Methods inherited from class org.dice_research.topicmodeling.io.AbstractModelFilesWriter

getFolder, setFolder

Methods inherited from class java.lang.Object

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

Field Detail

logger

private static final org.slf4j.Logger logger

TOPIC_PROBS_FILE_NAME

private static final String TOPIC_PROBS_FILE_NAME

See Also:

Constant Field Values

TOPIC_PROBS_FILE_HEAD

private static final String[] TOPIC_PROBS_FILE_HEAD

WORD_TOPIC_PROBS_FILE_NAME

private static final String WORD_TOPIC_PROBS_FILE_NAME

See Also:

Constant Field Values

WORD_TOPIC_SEPARATING_PROBS_FILE_NAME

private static final String WORD_TOPIC_SEPARATING_PROBS_FILE_NAME

See Also:

Constant Field Values

TOP_WORDS_FILE

private static final String TOP_WORDS_FILE

See Also:

Constant Field Values

SEPARATING_TOP_WORDS_FILE

private static final String SEPARATING_TOP_WORDS_FILE

See Also:

Constant Field Values

numberOfTopWordsPerTopic

protected int numberOfTopWordsPerTopic

TOPICS_FOR_DOCUMENTS_FILE

private static final String TOPICS_FOR_DOCUMENTS_FILE

See Also:

Constant Field Values

numberOfTopTopicsPerDocument

protected int numberOfTopTopicsPerDocument

fileNamePrefix

protected String fileNamePrefix

formatter

protected NumberFormat formatter

probTopicStateSupplier

protected org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier probTopicStateSupplier

printWordTopicProbs

protected boolean printWordTopicProbs

Constructor Detail

ModelCSVWriter

public ModelCSVWriter(org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier probTopicStateSupplier,
                      File folder,
                      boolean printWordTopicProbs)

ModelCSVWriter

public ModelCSVWriter(org.dice_research.topicmodeling.algorithms.ProbTopicModelingAlgorithmStateSupplier probTopicStateSupplier,
                      File folder)

ModelCSVWriter

public ModelCSVWriter(File folder,
                      String fileNamePrefix)

Method Detail

writeModelToFiles

public void writeModelToFiles(org.dice_research.topicmodeling.algorithms.Model model)

Specified by:

writeModelToFiles in interface ModelWriter

setNumberOfTopWordsPerTopic

public void setNumberOfTopWordsPerTopic(int numberOfTopWordsPerTopic)

writeModelToCSVFile

protected void writeModelToCSVFile(org.dice_research.topicmodeling.algorithms.Model model)

writeProbabilisticWordTopicModel

protected void writeProbabilisticWordTopicModel(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)

writeTopicFile

protected void writeTopicFile(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)

processWordTopicProbabilities

protected ModelCSVWriter.TopWordContainer processWordTopicProbabilities(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)

writeTopWords

protected void writeTopWords(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model,
                             ModelCSVWriter.TopWordContainer topWords)

processSeparatingWordTopicProbabilities

protected ModelCSVWriter.TopWordContainer processSeparatingWordTopicProbabilities(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)

writeSeparatingTopWords

protected void writeSeparatingTopWords(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model,
                                       ModelCSVWriter.TopWordContainer topWords)

writeTopicsForDocuments

protected void writeTopicsForDocuments(org.dice_research.topicmodeling.algorithms.ProbabilisticWordTopicModel model)