java - 如何在 Java 中加载 Weka 模型？

Question

我通过右键单击模型并选择“保存模型”来保存 Weka 分类的结果。现在，我想加载它并在我的 Java 应用程序中使用它。我怎样才能做到这一点？模型可以是朴素贝叶斯、决策树或回归。我需要使用这三个模型。

任何建议或解决方案将不胜感激。

Answer 1

这是一个示例，假设您将 RandomTree 模型保存到model.weka文件中（更改为您拥有的任何分类器和文件）

RandomTree treeClassifier = (RandomTree) SerializationHelper.read(new FileInputStream("model.weka")));

Answer 2

如果您将模型保存到 WEKA 中的文件中，则可以使用它来读取生成的 java 对象。这是一个带有随机森林分类器的示例（之前保存到 WEKA 中的文件）：

RandomForest rf = (RandomForest) (new ObjectInputStream(PATH_TO_MODEL_FILE)).readObject();

不要忘记进口：

import weka.core.Instance;
import weka.core.Instances;
import weka.core.Attribute;
import weka.core.FastVector;
import weka.classifiers.trees.RandomForest;