如何获取特定主题的单词概率分布?

时间:2015-01-15 13:43:17

标签: machine-learning nlp topic-modeling mallet

我正在使用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");

0 个答案:

没有答案