public void WeirdColumnTest() { Stream weirdStream = TestUtils.RetrieveResource(weirdColumnResource); var weirdConfig = GetWeirdConfig(); CCRecordSet recordSet = CCRecordSet.FromStream(weirdStream, weirdConfig); var nbc = ClassifierFactory.GetClassifierByName <string>("NaiveBayesClassifier"); nbc.Train(recordSet.Select(rec => new KeyValuePair <string, string>(rec.Description, rec.PredictedValues["Fish"])).ToList()); Assert.AreEqual(nbc.Categorize("Seattle Lounge").Category, "Trout"); }
public void NaiveBayesClassificationIntegrationTest() { Stream oldRecordsStream = TestUtils.RetrieveResource(fullChargeList); CCRecordSet records = CCRecordSet.FromStream(oldRecordsStream, config); var nbc = ClassifierFactory.GetClassifierByName <string>("NaiveBayesClassifier"); var trainingData = records .Select(rec => new KeyValuePair <string, string>(rec.Description, rec.PredictedValues["Category"])) .ToList(); nbc.Train(trainingData); Assert.AreEqual(nbc.Categorize("Trader Joe's").Category, "GROC"); Assert.AreEqual(nbc.Categorize("Shell Oil 27440482209 Seattle Wa").Category, "TRANS"); trainingData = records .Select(rec => new KeyValuePair <string, string>(rec.Description, rec.PredictedValues["Owner"])) .ToList(); nbc = ClassifierFactory.GetClassifierByName <string>("NaiveBayesClassifier"); nbc.Train(trainingData); Assert.AreEqual(nbc.Categorize("Radio Shack 00133652 Knoxville").Category, "Bob"); }