Exemplo n.º 1
0
        public void test_making_single_predictions_from_trained_model()
        {
            Runtime      rt         = Runtime.LoadFromFile <TitanicDataRow>(0, TestingHelpers.GetResourceFileName("titanic_train.csv"));
            RandomForest classifier = rt.Classifiers.Trees.RandomForest.
                                      NumExecutionSlots(4).
                                      NumFeatures(5).
                                      NumTrees(50);

            TitanicDataRow row = new TitanicDataRow
            {
                age      = 10,
                pclass   = "1",
                sex      = "male",
                embarked = "C"
            };
            double prediction = classifier.ClassifyRow(row);
            double proba      = classifier.ClassifyRowProba(row);

            Assert.AreEqual(0, prediction);
            Assert.IsTrue(proba < 0.5);
        }