Beispiel #1
0
        public void PlayGolfExample()
        {
            //http://www.saedsayad.com/decision_tree.htm
            var outlook  = new int[] { 0, 0, 1, 2, 2, 2, 1, 0, 0, 2, 0, 1, 1, 2 };
            var temp     = new int[] { 0, 0, 0, 1, 2, 2, 2, 1, 2, 1, 1, 1, 0, 1 };
            var humidity = new int[] { 0, 0, 0, 0, 1, 1, 1, 0, 1, 1, 1, 0, 1, 0 };
            var wind     = new int[] { 0, 1, 0, 0, 0, 1, 1, 0, 0, 0, 1, 1, 0, 1 };
            var output   = new int[] { 0, 0, 1, 1, 1, 0, 1, 0, 1, 1, 1, 1, 1, 0 };
            DecisionTreeLearning <int> dtl = DecisionTreeLearning <int> .Create <int>(x => output[x], new DecisionTreeOptions()
            {
                MaxDepth = 3,
            });

            var outlookStrs = new string[] { "Rainy", "Overcast", "Sunny" };

            dtl.AddDiscreteFeature <string>("outlook", a => outlookStrs[outlook[a]]);
            var tempStrs = new string[] { "Hot", "Mild", "Cool" };

            dtl.AddDiscreteFeature <string>("temp", a => tempStrs[temp[a]]);
            dtl.AddDiscreteFeature <int>("humidity", a => humidity[a]);
            dtl.AddDiscreteFeature <int>("wind", a => wind[a]);
            var dt = dtl.Learn(Enumerable.Range(0, output.Length));

            Assert.AreEqual(dt.Feature.Name, "outlook");
            Assert.AreEqual(dt.ChildNodes.First(x => (x.Key as DiscreteFeatureValue).Value.ToString() == "Rainy").Value.Feature.Name, "humidity");
            Assert.AreEqual(dt.ChildNodes.First(x => (x.Key as DiscreteFeatureValue).Value.ToString() == "Sunny").Value.Feature.Name, "wind");
            Assert.IsNull(dt.ChildNodes.First(x => (x.Key as DiscreteFeatureValue).Value.ToString() == "Overcast").Value.Feature, "Overcast node should be leaf");

            Assert.AreEqual(dt.GetOutput(0), 0);
            Assert.AreEqual(dt.GetOutput(2), 1);
        }
Beispiel #2
0
        public void ContiniousFeatureTest()
        {
            var data = Enumerable.Range(0, 1000).Select(x => new PatientRecord()).ToArray();

            var dt = DecisionTreeLearning <PatientRecord> .Create <int>(x => x.Cancer? 1 : 0, new DecisionTreeOptions());

            dt.AddContiniousFeature("Age", x => x.Age);
            dt.AddContiniousFeature("Height", x => x.Height);
            dt.AddContiniousFeature("Weight", x => x.Weight);
            dt.AddDiscreteFeature("Smoke", x => x.Smoke ? 1 : 0);
            var node = dt.Learn(data);

            Assert.AreEqual(node.Feature.Name, "Smoke");
        }