testnaivebayes.java

来自「一个使用的搜索引擎」· Java 代码 · 共 34 行

package ir.classifiers;

import java.util.*;

/**
 * Wrapper class to test NaiveBayes classifier using 10-fold CV.
 * Running it with -debug option gives very detailed output
 *
 * @author       Sugato Basu
 */

public class TestNaiveBayes {
    public static void main(String args[]) throws Exception
    {
	String dirName = "/u/mooney/ir-code/corpora/yahoo-science/";
	String[] categories = {"bio","chem","phys"};
	System.out.println("\nLoading Examples ...");
	List examples = new DirectoryExamplesConstructor(dirName, categories).getExamples();
	System.out.println("\nInitializing Naive Bayes classifier ...");
	NaiveBayes BC;
	boolean debug;
	// setting debug flag gives very detailed output, suitable for debugging
	if (args.length==1 && args[0].equals("-debug"))
	    debug = true;
	else
	    debug = false;
	BC = new NaiveBayes(categories, debug);

	// Perform 10-fold cross validation to generate learning curve
	CVLearningCurve cvCurve = new CVLearningCurve(BC,examples);
	cvCurve.run();
    }
}