我正在使用 Mallet 进行主题建模,一切正常,除了我无法获得任何特定主题中单词的概率分布。
但是,我正在使用下面的代码来打印任何特定文档的主题比例(由下面的 docID 变量表示) - 是否有任何类似的代码可以获取 Mallet 中任何特定主题的单词分布?
for (int topic = 0; topic < numTopics; topic++) {
Iterator<IDSorter> iterator = topicSortedWords.get(topic).iterator();
out = new Formatter(new StringBuilder(), Locale.US);
out.format("%d\t%.3f\t", topic, model.getTopicProbabilities(docID)[topic]);
System.out.println(out);
}
System.out.println("\n");