-
Notifications
You must be signed in to change notification settings - Fork 12
Expand file tree
/
Copy pathGraphFeatGen.java
More file actions
80 lines (66 loc) · 2.64 KB
/
Copy pathGraphFeatGen.java
File metadata and controls
80 lines (66 loc) · 2.64 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
package graph;
import edu.illinois.cs.cogcomp.sl.core.AbstractFeatureGenerator;
import edu.illinois.cs.cogcomp.sl.core.IInstance;
import edu.illinois.cs.cogcomp.sl.core.IStructure;
import edu.illinois.cs.cogcomp.sl.util.IFeatureVector;
import edu.illinois.cs.cogcomp.sl.util.Lexiconer;
import rate.RateFeatGen;
import rate.RateX;
import run.RunFeatGen;
import run.RunX;
import utils.FeatGen;
import java.util.ArrayList;
import java.util.List;
public class GraphFeatGen extends AbstractFeatureGenerator {
public Lexiconer lm = null;
public GraphFeatGen(Lexiconer lm) {
this.lm = lm;
}
@Override
public IFeatureVector getFeatureVector(IInstance iInstance, IStructure iStructure) {
GraphX x = (GraphX) iInstance;
GraphY y = (GraphY) iStructure;
return FeatGen.getFeatureVectorFromListString(getFeatures(x, y), lm);
}
public IFeatureVector getRateFeatureVector(
GraphX x, int relevantQuantIndex, String label) {
return FeatGen.getFeatureVectorFromListString(
getRateFeatures(x, relevantQuantIndex, label), lm);
}
public IFeatureVector getRunFeatureVector(
GraphX x, int relevantQuantIndex1, int relevantQuantIndex2, String label) {
return FeatGen.getFeatureVectorFromListString(
getRunFeatures(x, relevantQuantIndex1, relevantQuantIndex2, label), lm);
}
public static List<String> getFeatures(GraphX x, GraphY y) {
List<String> feats = new ArrayList<>();
int n = x.relevantQuantIndices.size();
assert y.labels.size() == (n + n*(n-1)/2);
for(int i=0; i<n; ++i) {
feats.addAll(getRateFeatures(x, i, y.labels.get(i)));
}
int next = n;
for(int i=0; i<n; ++i) {
for(int j=i+1; j<n; ++j) {
feats.addAll(getRunFeatures(x, i, j, y.labels.get(next)));
next++;
}
}
return feats;
}
public static List<String> getRateFeatures(
GraphX x, int relevantQuantIndex, String label) {
List<String> feats = new ArrayList<>();
feats.addAll(RateFeatGen.getFeatures(new RateX(
x, x.relevantQuantIndices.get(relevantQuantIndex)), label));
return feats;
}
public static List<String> getRunFeatures(
GraphX x, int relevantQuantIndex1, int relevantQuantIndex2, String label) {
List<String> feats = new ArrayList<>();
feats.addAll(RunFeatGen.getFeatures(new RunX(
x, x.relevantQuantIndices.get(relevantQuantIndex1),
x.relevantQuantIndices.get(relevantQuantIndex2)), label));
return feats;
}
}