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

package org.greenstone.mars;

//
// Predict both Arousal and Valence values
//

import weka.core.Instances;

import weka.classifiers.Classifier;


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

class WekaApplyAVModels
{
    public static void main(String[] args)
    {
    WekaUtil.checkUsageApplyAVModels(args);
    
    String arousal_model_input_filename     = args[0];
    String valence_model_input_filename     = args[1];
    String unclassified_data_input_filename = args[2];
    String classified_data_output_filename  = args[3];

    Classifier arousal_classifier = WekaUtil.loadClassifierModel(arousal_model_input_filename);
    Classifier valence_classifier = WekaUtil.loadClassifierModel(valence_model_input_filename);

    Instances unlabeled_instances= WekaUtil.loadInstancesForClassification(unclassified_data_input_filename);

    Instances filtered_unlabeled_instances = WekaUtil.applyFilter(unlabeled_instances,null); // no additional top-up to remove
            
    WekaUtil.appendUnclassifiedAttribute(filtered_unlabeled_instances,WekaUtil.AROUSAL_ATTRIBUTE_NAME);
    System.out.println("Predicting arousal:");
    Instances arousal_labeled_instances = WekaUtil.makePredictions(arousal_classifier, filtered_unlabeled_instances, null);

    WekaUtil.appendUnclassifiedAttribute(arousal_labeled_instances,WekaUtil.VALENCE_ATTRIBUTE_NAME);
    System.out.println("Predicting valence:");
    Instances av_labeled_instances = WekaUtil.makePredictions(valence_classifier, arousal_labeled_instances, null);

    /*
    try {
        // Save labeled data
        
        System.out.println("Saving labeled instances: " + classified_data_output_filename);     
        FileWriter fw = new FileWriter(classified_data_output_filename);
        BufferedWriter bw = new BufferedWriter(fw);
        
        bw.write(av_labeled_instances.toString());
        bw.newLine();
        bw.flush();
        bw.close();
             
    }
    catch (Exception e) {
        e.printStackTrace();
    }
    */

    /*
    try {
        System.out.println("Saving labeled instances: " + classified_data_output_filename);     
        DataSink.write(classified_data_output_filename, av_labeled_instances);
    }
    catch (Exception e) {
        System.err.println("Failed to save data to: " + classified_data_output_filename);
        e.printStackTrace();
    }
    */
    
    WekaUtil.saveInstancesAsDataSink(av_labeled_instances,classified_data_output_filename);
        
    }
}