|
public train ( |
||
| ds | ||
| return | void | |
public void testDecisionListTestRunOnRestaurantDataSet()
{
DataSet ds = DataSetFactory.getRestaurantDataSet();
DecisionListLearner learner = new DecisionListLearner("Yes", "No",
new DLTestFactory());
learner.train(ds);
int[] result = learner.test(ds);
Assert.AreEqual(12, result[0]);
Assert.AreEqual(0, result[1]);
}
public void testDecisionListLearnerReturnsNegativeDLWhenDataSetEmpty()
{
// tests first base case of DL Learner
DecisionListLearner learner = new DecisionListLearner("Yes", "No",
new MockDLTestFactory(null));
DataSet ds = DataSetFactory.getRestaurantDataSet();
DataSet empty = ds.emptyDataSet();
learner.train(empty);
Assert.AreEqual("No", learner.predict(ds.getExample(0)));
Assert.AreEqual("No", learner.predict(ds.getExample(1)));
Assert.AreEqual("No", learner.predict(ds.getExample(2)));
}
public void testDecisionListLearnerReturnsFailureWhenTestsEmpty()
{
// tests second base case of DL Learner
DecisionListLearner learner = new DecisionListLearner("Yes", "No",
new MockDLTestFactory(new List<DLTest>()));
DataSet ds = DataSetFactory.getRestaurantDataSet();
learner.train(ds);
Assert.AreEqual(DecisionListLearner.FAILURE, learner.predict(ds
.getExample(0)));
}
