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

Introduction

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

Prototype

@Override
public Instances defineDataFormat() throws Exception 

Source Link

Document

Initializes the format for the dataset produced.

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./*from www .j  a  va 2 s.  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();
}

---