SplitNoClass.java

/**
 * The program is for Active Learning Challenge.
 */
package weka.filters;

import java.io.*;
import java.util.*;
import weka.core.*;

public class SplitNoClass extends Filter{
  public static void filterFile(Filter filter, String [] options) 
    throws Exception {
    String data,label,sparse;
    Instances mydata = null;
    BufferedReader dat = null;
    BufferedReader lab = null;
    boolean isSparse=false, isFind = false;
    String line = "";

    try {
      data=Utils.getOption('d',options);
      label=Utils.getOption('l',options);
      sparse=Utils.getOption('s',options);

      if (data.length() != 0) {
	dat = new BufferedReader(new FileReader(data));
      } else {
	dat = new BufferedReader(new InputStreamReader(System.in));
      }

      if (label.length() != 0) {
	lab = new BufferedReader(new FileReader(label));
      } else {
	lab = new BufferedReader(new InputStreamReader(System.in));
      }

      if(sparse.length()!=0){
          isSparse=true;
      }

      mydata = new Instances(dat);
      mydata.setClassIndex(mydata.numAttributes()-1);

    } catch (Exception ex) {
      String genericOptions = "\nGeneral options:\n\n"
	+ "-h\n"
	+ "\tGet help on available options.\n"
	+ "\t(use -b -h for help on batch mode.)\n"
	+ "-d <file>\n"
	+ "\tThe name of the file containing input instances with class attribute but no value.\n"
	+ "\tIf not supplied then instances will be read from stdin.\n"
	+ "-l <file>\n"
	+ "\tThe name of the file containing input query label.\n"
	+ "-s <is sparse>\n";
      throw new Exception('\n' + ex.getMessage()
			  +genericOptions);
    }
    
    String path = label.substring(0,label.lastIndexOf("."));
    PrintWriter output1 = new PrintWriter(new FileOutputStream(path+"-train.arff"));
    PrintWriter output2 = new PrintWriter(new FileOutputStream(path+"-test.arff"));

    Instances train=new Instances(mydata,0);
    train.setRelationName(mydata.relationName()+".train");
    Instances test=new Instances(mydata,0);
    test.setRelationName(mydata.relationName()+".test");
    Instance instance;

output1.println(train.toString());
output2.println(test.toString());

ArrayList qr = new ArrayList();
while ( (line = lab.readLine()) != null) {
		String[] temp = line.split("\\s+");
		int[] temp1 = new int[2];
		temp1[0] = Integer.parseInt(temp[0])-1;
		temp1[1] = Integer.parseInt(temp[1]);
                qr.add(temp1);
}

for(int i = 0; i < mydata.numInstances(); i ++){
	instance = mydata.instance(i); 
	isFind = false;
	for(int j = 0; j < qr.size(); j ++){
		int[] temp3 = (int[])(qr.get(j));
		if(temp3[0] == i){
			instance.setClassValue(Integer.toString(temp3[1]));
			isFind = true; 
			break;
		}
    	}
	if(isSparse){ instance = new SparseInstance(instance); }
	if(isFind == false){ output2.println(instance.toString());System.out.println(i+1);}
	if(isFind == true) { output1.println(instance.toString());}
}

    if (output1 != null) {
      output1.close();
    }

    if (output2 != null) {
      output2.close();
    }
  }
  public static void main(String [] argv) {
    try {
	SplitNoClass.filterFile(new SplitNoClass(), argv);
    } catch (Exception ex) {
      System.out.println(ex.getMessage());
    }
  }
}