AIMA.Test.Core.Unit.Learning.Learners.DecisionTreeTest.testStumpPredictionForDataSet C# (CSharp) Method

        public void testStumpPredictionForDataSet()
        {
            DataSet ds = DataSetFactory.getRestaurantDataSet();

            List<String> unmatchedValues = new List<String>();
            unmatchedValues.Add(NO);
            DecisionTree tree = DecisionTree.getStumpFor(ds, "hungry", YES, YES,
                    unmatchedValues, "Unable to Classify");
            DecisionTreeLearner learner = new DecisionTreeLearner(tree,
                    "Unable to Classify");
            int[] result = learner.test(ds);
            Assert.AreEqual(5, result[0]);
            Assert.AreEqual(7, result[1]);
        }