use of weka.classifiers.functions.SMOreg in project dkpro-tc by dkpro.
the class WekaResultsTest method testWekaResultsRegression.
@Test
public void testWekaResultsRegression() throws Exception {
SMOreg cl = new SMOreg();
Instances trainData = WekaUtils.removeInstanceId(regressionTrainData, false);
Instances testData = WekaUtils.removeInstanceId(regressionTestData, false);
cl.buildClassifier(trainData);
Evaluation eval = WekaUtils.getEvaluationSinglelabel(cl, trainData, testData);
assertEquals(0.45, eval.correlationCoefficient(), 0.01);
}
Aggregations