source: main/trunk/model-sites-dev/mars/src/java/org/greenstone/mars/WekaTrainValenceModel.java@ 34797

package org.greenstone.mars;
    
//import weka.core.converters.ConverterUtils.DataSource;
import weka.core.Instances;
 
//import weka.filters.Filter;
//import weka.filters.unsupervised.attribute.Remove;

import weka.classifiers.Classifier;
import weka.classifiers.Evaluation;
//import weka.classifiers.trees.REPTree;

// Based on:
//   https://waikato.github.io/weka-wiki/use_weka_in_your_java_code/


class WekaTrainValenceModel
{
    // Scheme:       weka.classifiers.trees.REPTree -M 2 -V 0.001 -N 3 -S 1 -L -1 -I 0.0   -- (i.e., default vals)
    // Relation:     deam-essentia-features-arousal-valence-weka.filters.unsupervised.attribute.Remove-R471-weka.filters.unsupervised.attribute.Remove-R458-466-weka.filters.unsupervised.attribute.Remove-R425
    // Instances:    1743

    // Note, above missed -R 427 (ordinal val)

    
    public static void main(String[] args)
    {
    /*
    if (args.length != 2) {
        System.err.println("Error: incorrect number of command-line arguments");
        System.err.println("Usage: input_training_data.arff output-model.{model|ser}");
        System.exit(1);
        }*/
    
    WekaUtil.checkUsageTraining(args);

    String input_data_filename = args[0];
    String output_model_filename = args[1];

    // knock out 471 (arousal) from dataset
    Instances filtered_data_instances = WekaUtil.loadAndFilterDataInstances(input_data_filename, "471");
    Classifier classifier = WekaUtil.trainREPTree(filtered_data_instances);
    System.out.println(classifier);
                        
    Evaluation eval = WekaUtil.evaluateClassifier(classifier,filtered_data_instances);
    System.out.println(eval.toSummaryString());

    try {
        System.out.println("Saving REPTree classifier model as: " + output_model_filename);
        weka.core.SerializationHelper.write(output_model_filename, classifier);
    }
    catch (Exception e) {
        e.printStackTrace();
    }

    /*  
        System.out.println("Training on ARFF file: " + input_arff_filename);

    try {
        DataSource data_source = new DataSource(input_arff_filename);
        Instances data_instances = data_source.getDataSet();

        // *********
        Instances filtered_data_instances = WekaUtil.applyFilter(data_instances,"471"); // top-up with removal of 'arousal'

        // ****** comment varies
        // With removal of R471, last column is the one we want to predict => numAttributes()-1
        filtered_data_instances.setClassIndex(filtered_data_instances.numAttributes()-1);
        

        // Build scheme/classifier
        REPTree classifier = new REPTree(); // scheme
        String[] reptree_options = weka.core.Utils.splitOptions("-M 2 -V 0.001 -N 3 -S 1 -L -1 -I 0.0");
        classifier.setOptions(reptree_options);

        classifier.buildClassifier(filtered_data_instances);
        System.out.println(classifier);

        
        Evaluation eval = WekaUtil.evaluateClassifier(classifier,filtered_data_instances);
        System.out.println(eval.toSummaryString());
            
        System.out.println("Saving REPTree classifier model as: " + output_model_filename);
        weka.core.SerializationHelper.write(output_model_filename, classifier);

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