public void TestGetWordsDataSource()
{
SimpleWordsDataSource wds = new SimpleWordsDataSource();
BayesianClassifier classifier = new BayesianClassifier(wds);
Assert.AreEqual(wds, classifier.WordsDataSource);
}
public void TestGetTokenizer()
{
SimpleWordsDataSource wds = new SimpleWordsDataSource();
ITokenizer tokenizer = new DefaultTokenizer(DefaultTokenizer.BREAK_ON_WORD_BREAKS);
BayesianClassifier classifier = new BayesianClassifier(wds, tokenizer);
Assert.AreEqual(tokenizer, classifier.Tokenizer);
}
public void TestGetStopWordProvider()
{
var wds = new SimpleWordsDataSource();
ITokenizer tokenizer = new DefaultTokenizer(DefaultTokenizer.BREAK_ON_WORD_BREAKS);
IStopWordProvider stopWordProvider = new DefaultStopWordProvider();
var classifier = new BayesianClassifier(wds, tokenizer, stopWordProvider);
Assert.AreEqual(stopWordProvider, classifier.StopWordProvider);
}
public void TestClassify()
{
SimpleWordsDataSource wds = new SimpleWordsDataSource();
BayesianClassifier classifier = new BayesianClassifier(wds);
string[] sentence = new string[] { "This", "is", "a", "sentence", "about", "java" };
Assert.AreEqual(IClassifierConstants.NEUTRAL_PROBABILITY, classifier.Classify(ICategorizedClassifierConstants.DEFAULT_CATEGORY, sentence), 0d);
wds.SetWordProbability(new WordProbability("This", .5d));
wds.SetWordProbability(new WordProbability("is", .5d));
wds.SetWordProbability(new WordProbability("a", .5d));
wds.SetWordProbability(new WordProbability("sentence", .2d));
wds.SetWordProbability(new WordProbability("about", .5d));
wds.SetWordProbability(new WordProbability("java", .99d));
Assert.AreEqual(.96d, classifier.Classify(ICategorizedClassifierConstants.DEFAULT_CATEGORY, sentence), .009d);
}
protected void Setup()
{
wordsDataSource = new SimpleWordsDataSource();
}
