public test ( |
||
ds | ||
return | int[] |
public int[] test(DataSet ds)
{
int[] results = new int[] { 0, 0 };
foreach (Example e in ds.examples)
{
if (e.targetValue().Equals(tree.predict(e)))
{
results[0] = results[0] + 1;
}
else
{
results[1] = results[1] + 1;
}
}
return results;
}
public void testInducedDecisionTreeClassifiesRestaurantDataSetCorrectly() { DecisionTreeLearner learner = new DecisionTreeLearner( createInducedRestaurantDecisionTree(), "Unable to clasify"); int[] results = learner.test(DataSetFactory.getRestaurantDataSet()); Assert.AreEqual(12, results[0]); Assert.AreEqual(0, results[1]); }