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");
}