public void NaiveBayersSpamTrueTest()
{
NaiveBayes.NaiveBayes naiveBayes = new NaiveBayes.NaiveBayes();
var result = naiveBayes.CheckEmail("Buy Cheap cialis Online");
Assert.AreEqual(true, result);
result = naiveBayes.CheckEmail("Enlarge Your Penis");
Assert.AreEqual(true, result);
result = naiveBayes.CheckEmail("accept VISA, MasterCard");
Assert.AreEqual(true, result);
}
public void NaiveBayersNotSpamFalseTest()
{
NaiveBayes.NaiveBayes naiveBayes = new NaiveBayes.NaiveBayes();
var result = naiveBayes.CheckEmail("Sincerely Mathias");
Assert.AreEqual(false, result);
result = naiveBayes.CheckEmail("Dear Graduates");
Assert.AreEqual(false, result);
result = naiveBayes.CheckEmail("Thanks in advance for your support");
Assert.AreEqual(false, result);
result = naiveBayes.CheckEmail("for it with my Mastercard");
Assert.AreEqual(false, result);
}
private void loadDataToolStripMenuItem_Click(object sender, EventArgs e)
{
var start = System.DateTime.Now;
IFileLoader fileLoader = new FileLoader.FileLoader();
var arffFilePath = fileLoader.GetPath(string.Empty);
IFileProcesser<string> fileProcesser = new SimpleFileProcceser();
using (var textReader = new StreamReader(arffFilePath, Encoding.UTF8))
{
var processedFile = fileProcesser.Process(textReader.ReadToEnd());
var naiveBayes = new NaiveBayes.NaiveBayes(processedFile, "Drug");
naiveBayes.TrainFromSet();
fileLoader.SaveJsonFileToText(JsonConvert.SerializeObject(naiveBayes.GetModel()), string.Empty);
}
var end = System.DateTime.Now;
var difference = end.Subtract(start).TotalMilliseconds;
MessageBox.Show(@"Processed! in " + difference + " milliseconds", @"Time elapsed", MessageBoxButtons.OK);
}
