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