diff --git a/BotSharp.NLP.UnitTest/NaiveBayesClassifierTest.cs b/BotSharp.NLP.UnitTest/NaiveBayesClassifierTest.cs index cffd7c00..be599c14 100644 --- a/BotSharp.NLP.UnitTest/NaiveBayesClassifierTest.cs +++ b/BotSharp.NLP.UnitTest/NaiveBayesClassifierTest.cs @@ -34,7 +34,7 @@ namespace BotSharp.NLP.UnitTest } sentences = newSentences.ToList(); - // sentences.Shuffle(); + sentences.Shuffle(); var encoder = new OneHotEncoder(); encoder.Sentences = sentences; @@ -46,23 +46,24 @@ namespace BotSharp.NLP.UnitTest }; var classifier = new ClassifierFactory(options, SupportedLanguage.English); - var dataset = sentences.Split(1M); + var dataset = sentences.Split(0.7M); classifier.Train(dataset.Item1); int correct = 0; - dataset.Item1.ToList().ForEach(td => + int total = 0; + dataset.Item1.ForEach(td => { var classes = classifier.Classify(td); if (td.Label == classes[0].Item1) { correct++; - } + total++; }); - var accuracy = (float)correct / dataset.Item1.Count; + var accuracy = (float)correct / total; - Assert.IsTrue(accuracy > 0.8); + Assert.IsTrue(accuracy > 0.5); } [TestMethod]