source: trunk/CrossPare/src/de/ugoe/cs/cpdp/training/RandomForestTraining.java @ 10

package de.ugoe.cs.cpdp.training;

import weka.classifiers.Classifier;
import weka.classifiers.meta.CVParameterSelection;
import weka.classifiers.trees.RandomForest;

public class RandomForestTraining extends WekaTraining {

        @Override
        protected Classifier setupClassifier() {
                final CVParameterSelection ps = new CVParameterSelection();
                ps.setClassifier(new RandomForest());
                try {
                        // Parameter optimization with 5x5 CV
                        ps.setNumFolds(5);
                        ps.addCVParameter("I 5 25 5");
                } catch (Exception e) {
                        throw new RuntimeException(e);
                }
                return ps;
        }
        
        @Override
        public String getName() {
                return "RandomForest";
        }

}