de.tudarmstadt.ukp.dkpro.core.mallet.lda

Class MalletLdaTopicModelUtils

java.lang.Object

de.tudarmstadt.ukp.dkpro.core.mallet.lda.MalletLdaTopicModelUtils

public class MalletLdaTopicModelUtils
extends Object

Constructor Summary

Constructors

Constructor and Description
MalletLdaTopicModelUtils()

Method Summary

Modifier and Type	Method and Description
static List<Map<String,Double>>	getTopWords(File modelFile, int nWords, boolean normalize) Retrieve the top n topic words for each topic in the given model.
static void	printTopicWords(File modelFile, File targetFile, int nWords) Print the top n words of each topic into a file.

Methods inherited from class java.lang.Object

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

Constructor Detail

MalletLdaTopicModelUtils

public MalletLdaTopicModelUtils()

Method Detail

getTopWords

public static List<Map<String,Double>> getTopWords(File modelFile,
                                                   int nWords,
                                                   boolean normalize)
                                            throws IOException

Retrieve the top n topic words for each topic in the given model.

Parameters:

modelFile - the model file

nWords - the maximum number of words to retrieve

normalize - normalize the word weights ?

Returns:

a list of maps where each map represents a topic, mapping words to weights

Throws:

IOException - if the model cannot be read

printTopicWords

public static void printTopicWords(File modelFile,
                                   File targetFile,
                                   int nWords)
                            throws IOException

Print the top n words of each topic into a file.

Parameters:

modelFile - the model file

targetFile - the file in which the topic words are written

nWords - the number of words to extract

Throws:

IOException - if the model file cannot be read or if the target file cannot be written