使用Java进行Weka预测

时间:2015-11-08 07:54:11

标签: java weka

我需要在Java中使用回归来预测结束sem标记。这就是我到目前为止所拥有的

public class LR{
    public static void main(String[] args) throws Exception
    {
        BufferedReader datafile = new BufferedReader(new FileReader("C:\\dataset.arff"));
        Instances data = new Instances(datafile);

        data.setClassIndex(data.numAttributes()-1);    //setting class attribute
        datafile.close();

        LinearRegression lr = new LinearRegression();  //build model
        int folds=10;

        Evaluation eval = new Evaluation(data);
        eval.crossValidateModel(lr, data, folds, new Random(1));
        System.out.println(eval.toSummaryString()); 

        //save the model
        weka.core.SerializationHelper.write("C:\\lr.model", lr);

        //load the model
        Classifier cls = (Classifier)weka.core.SerializationHelper.read("C:\\lr.model");

    }
}  

如何使用加载的模型预测另一个文件中学生的标记:"testfile"

1 个答案:

答案 0 :(得分:0)

将另一个文件加载为Weka实例,遍历每个实例并调用cls.classifyInstance

请参阅This example