-
Notifications
You must be signed in to change notification settings - Fork 0
Expand file tree
/
Copy pathProgram.java
More file actions
105 lines (85 loc) · 2.43 KB
/
Program.java
File metadata and controls
105 lines (85 loc) · 2.43 KB
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
87
88
89
90
91
92
93
94
95
96
97
98
99
100
101
102
103
104
105
import java.io.*;
import java.util.ArrayList;
import java.util.List;
import java.util.Random;
public class Program {
/**
* @param args
*/
public static void main(String[] args) {
System.err.println("Usage: java Ensemble scoreFile_1:scoreFile_2:... weight_1:weight_2:... output_ensemble_score_file");
try {
if (args.length != 3) {
System.err.println("wrong input arguments - exit");
System.exit(-1);
}
String[] scoreFiles = args[0].split(":");
String[] vals = args[1].split(":");
float[] weights = new float[vals.length];
int j = 0;
for (String e : vals) {
weights[j++] = Float.parseFloat(e);
}
BufferedReader[] brs = new BufferedReader[vals.length];
for (int i=0; i<scoreFiles.length; i++) {
brs[i] = new BufferedReader(new FileReader(scoreFiles[i]));
}
PrintWriter pw = new PrintWriter(new File(args[2]));
String line = brs[0].readLine(); // first line;
vals = line.split("\t");
int pos = -1;
for (int i=0; i<vals.length; i++) {
if (vals[i].equals("m:Predicted")) {
pos = i;
break;
}
}
for (int i=1; i<brs.length; i++) {
brs[i].readLine();
}
pw.println(line);
int cnt = 0;
float[] scores = new float[brs.length];
while ((line = brs[0].readLine()) != null) {
vals = line.split("\t");
scores[0] = Float.parseFloat(vals[pos]);
for (int i=1; i<brs.length; i++) {
line = brs[i].readLine();
String[] temp = line.split("\t");
scores[i] = Float.parseFloat(temp[pos]);
}
float ensembleScore = ensemble(scores, weights);
StringBuilder sb = new StringBuilder();
for (int i=0; i<pos; i++) {
sb.append(vals[i] + "\t");
}
sb.append(ensembleScore + "\t");
for (int i=pos+1; i<vals.length; i++) {
sb.append(vals[i] + "\t");
}
pw.println(sb.toString().substring(0, sb.toString().length()-1));
cnt++;
if (cnt % 1000 == 0) {
System.err.println(cnt + " lines processed.");
}
}
pw.close();
for (int i=0; i<brs.length; i++) {
brs[i].close();
}
} catch (Exception ex) {
ex.printStackTrace();
}
}
public static float ensemble(float[] scores, float[] weights) {
if (scores.length != weights.length) {
System.err.println("wrong input arguments in ensemble() - exit");
System.exit(-1);
}
float sum = 0.0f;
for (int i=0; i<scores.length; i++) {
sum += weights[i] * scores[i];
}
return sum;
}
}