【发布时间】:2016-01-20 07:48:38
【问题描述】:
我对 Weka 还很陌生。 我正在编写一个代码,其中我构建了一个 J48 来预测学生成绩的结果。 现在我尝试使用 ARFF 文件测试模型,但我想实现用户输入的测试用例的分类。 例如。我希望用户输入两个数值,它们是在两个科目(即 CS 和数学)中获得的分数,然后预测他们的最终结果,即通过或失败。结果将是类变量。
我不知道如何创建一个实例来执行这样的事情
double pred = tree.ClassifyInstance(testcase);
这是我的代码。
import java.security.KeyStore;
import weka.classifiers.Classifier;
import weka.classifiers.Evaluation;
import weka.classifiers.trees.J48;
import weka.core.Attribute;
import weka.core.FastVector;
import weka.core.Instance;
import weka.core.Instances;
import weka.core.Utils;
import weka.core.converters.ConverterUtils.DataSource;
public class WekaTest {
public static void main (String[] args) throws Exception
{
//load the train set
DataSource source = new DataSource("train.arff");
Instances train = source.getDataSet();
//Set class variable i.e. Outcome
train.setClassIndex(train.numAttributes()- 1);
Classifier tree = new J48();
tree.buildClassifier(train);
// Instance test = new Instance();
// Evaluation eval = new Evaluation(train);
Attribute COMS1000 = new Attribute("COMS1000");
Attribute MATH1001 = new Attribute("MATH1001");
FastVector classVal = new FastVector(2);
classVal.addElement("PASS");
classVal.addElement("FAIL");
FastVector testAttributes = new FastVector(3);
testAttributes.addElement(COMS1000);
testAttributes.addElement(MATH1001);
testAttributes.addElement(classVal);
Instance testcase = new Instance(3);
//testcase.setClassIndex(testcase.numAttributes()-1);
testcase.setValue((Attribute)testAttributes.elementAt(0),60);
testcase.setValue((Attribute)testAttributes.elementAt(1),70);
testcase.setValue((Attribute)testAttributes.elementAt(2),"?");
double pred = tree.classifyInstance(testcase)
System.out.println(pred.value(Double.toString(pred)));
}
}
【问题讨论】: