Ejemplo n.º 1
0
  public static void main(String[] args) throws IOException {
    Reader r = new Reader();
    ArrayList<String> labels = r.read("test.label");
    ArrayList<String> data = r.read("test.data");

    FileWriter fw = new FileWriter("test20.txt");
    for (int i = 1; i <= labels.size(); i++) {
      int label = Integer.parseInt(labels.get(i - 1));
      String doc = "" + (label - 1);
      while (!data.isEmpty()) {
        String s = data.get(0);
        String[] temp = s.split(" ");
        if (Integer.parseInt(temp[0]) != i) {
          break;
        }
        data.remove(0);
        String oneWord = " " + temp[1] + ":" + temp[2];
        doc = doc + oneWord;
      }
      doc = doc + "\r\n";
      fw.write(doc);
      System.out.println("doc " + i + "  finished");
    }
    fw.close();
  }
  public ArrayList<OneDoc> readText(String fileName) {

    ArrayList<OneDoc> result = new ArrayList<OneDoc>();

    Reader r = new Reader();
    ArrayList<String> stringList = r.read(fileName);
    for (String s : stringList) {
      result.add(new OneDoc(s));
    }

    return result;
  }