java - 如何将命令行训练的主题模型读入Java类？

Question

所以我有一个经过训练的模型，是通过命令行使用 MALLET 创建的。我想以某种方式将这个经过训练的模型导入到 Java 类中。我浏览了 MALLET 的 API 文档，发现了他们的 ParallelTopicModel 类，但找不到合适的方法将模型导入为 ParallelTopicModel，这正是我想要做的。

我知道可以在 Java 类本身内使用 MALLET Java API 来训练模型，但我不想这样做。

这可能吗？

Answer 1

如果经过训练的模型同时生成数据模型和数据实例，您就可以执行此操作。 ParallelTopicModel 将使用数据模型，并且用于训练模型的 Pipe 将用于创建您需要的 InstanceList对于TopicInferencer。

String testingText = "my test";
ParallelTopicModel model = ParallelTopicModel.read(new File("data_model_path"));
InstanceList instances = InstanceList.load(new File("data_instances_path"));

// Create a new instance with the testing text
InstanceList testing = new InstanceList(instances.getPipe());
testing.addThruPipe(new Instance(testing_text, null, "Test Instance", null));

// Create the inferencer
TopicInferencer inferencer = model.getInferencer();