SemianiakaVY/BotSharp
1
0
Fork 0
Code Issues Pull requests Actions Packages Projects Releases Wiki Activity

update CookingTest

Browse source
This commit is contained in:
Oceania2018 2018-09-11 17:46:07 -05:00
parent aec1984ab8
commit ee115ebde6
1 changed files with 7 additions and 6 deletions
Show stats Download patch file Download diff file Expand all files Collapse all files

13
BotSharp.NLP.UnitTest/NaiveBayesClassifierTest.cs
View file

@ -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<NaiveBayesClassifier, SentenceFeatureExtractor>(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]