naivebayesclassifiertest.java

来自「一个自然语言处理的Java开源工具包。LingPipe目前已有很丰富的功能」· Java 代码 · 共 26 行

package com.aliasi.test.unit.classify;

import com.aliasi.classify.NaiveBayesClassifier;

import com.aliasi.tokenizer.IndoEuropeanTokenizerFactory;

import com.aliasi.test.unit.BaseTestCase;

public class NaiveBayesClassifierTest extends BaseTestCase {

    public void testOne() {
    NaiveBayesClassifier classifier
        = new NaiveBayesClassifier(new String[] { "a", "b", "c" },
                       new IndoEuropeanTokenizerFactory());
    classifier.train("a","John Smith");
    classifier.train("a","John Smith");
    classifier.train("b","Fred Smith");
    classifier.train("b","Fred Smith");
    classifier.train("c","Fred Jones");
    classifier.train("c","Fred Jones");
    
    assertEquals("a",classifier.classify("John Smith").bestCategory());
    }

}