Example usage for weka.datagenerators.classifiers.classification Agrawal setBalanceClass

Introduction

In this page you can find the example usage for weka.datagenerators.classifiers.classification Agrawal setBalanceClass.

Prototype

public void setBalanceClass(boolean value) 

Source Link

Document

Sets whether the class is balanced.

Usage

From source file:NLP.GenerateData.java

License:Open Source License

/**
 * Generates the data. First parameter the ARFF file to write to. If omitted,
 * the data is written to stdout.//w w w. j  a v  a2s.  c  o  m
 * 
 * @param args   the command-line parameters
 */
public static void main(String[] args) throws Exception {
    PrintWriter output;
    if (args.length == 0)
        output = new PrintWriter(System.out);
    else
        output = new PrintWriter(new BufferedWriter(new FileWriter(args[0])));
    Agrawal generator = new Agrawal();
    generator.setBalanceClass(true);
    generator.setDatasetFormat(generator.defineDataFormat());
    // write header
    output.write(generator.getDatasetFormat().toString());
    // write data
    if (generator.getSingleModeFlag()) {
        for (int i = 0; i < generator.getNumExamplesAct(); i++) {
            output.write(generator.generateExample().toString());
            output.write(System.getProperty("line.separator"));
        }
    } else {
        output.write(generator.generateExamples().toString());
    }
    output.flush();
    output.close();
}

---