Пример #1
0
 public Machine(int numOfClasses, int numOfFeatures, int[] featureSelection, double noise)
     : base(numOfClasses, numOfFeatures, featureSelection, noise)
 {
     this.trainModel = new TrainModel(numOfClasses, noise);
     this.testModel = new TestModel(numOfClasses, noise);
 }
        public void TrainModelAndTestModelTest(string trainFile, string testFile)
        {
            var trainModel = new TrainModel(2, 0.1);
            var trainDataset = new ClassifiedDataset(trainFile, 64, 2);
            trainModel.Train(trainDataset.GetClassifiedVectors());

            var testModel = new TestModel(2, 0.1);
            var testDataset = new UnclassifiedDataset(testFile, 64);
            var testData = testDataset.GetDataVectors().Select(v => v.FeatureVector).ToArray();
            var results = testModel.Test(trainModel.GetInferredPosterier(), testData);

            Assert.That(results.Length, Is.EqualTo(testData.Length));
        }