java weka on java

Posted 2021-05-08

import weka.classifiers.Evaluation;
import weka.classifiers.bayes.BayesNet;
import weka.classifiers.bayes.net.search.SearchAlgorithm;
import weka.classifiers.bayes.net.search.local.SimulatedAnnealing;
import weka.core.Instances;
import weka.core.converters.ConverterUtils.DataSource;

public class TestWeka {

	public static void main(String[] args) {
		try {
		    // arffファイルを読み込む
		    DataSource source = new DataSource("werewolf1.arff");
		    Instances instances = source.getDataSet();
		    // 親ノードにしたいノードのインデックス？
		    instances.setClassIndex(0);

		    // BayesNet
		    BayesNet bnet = new BayesNet();
		    
		    // サーチアルゴリズムの生成
		    SearchAlgorithm searchAlgorithm=new SimulatedAnnealing();
		    //BayesNetにセット
		    bnet.setSearchAlgorithm(searchAlgorithm);
		    //分類開始
		    bnet.buildClassifier(instances);
		   
		    // 評価   
		    Evaluation evaluation = new Evaluation(instances);
		    evaluation.evaluateModel(bnet, instances);
		    System.out.println(evaluation.toSummaryString("Results\n",false));
		    
		    //ベイジアンネットワークのデータの取得
		    //(ノードインデックス,そのノードの値インデックス)
		    System.out.println(bnet.getNodeValue(1, 0));
		    //(ノードインデックス,その親ノードの値インデックス,そのノードの値インデックス)
		    System.out.println(bnet.getProbability(3, 2, 0));

		} catch (Exception e) {
			e.printStackTrace();
		}
	}
}