Skip to content
Snippets Groups Projects
Commit 09216e6e authored by Christos Christodoulopoulos's avatar Christos Christodoulopoulos
Browse files

Added sense description ("name") to FrameData

parent 80fd99e3
No related branches found
Tags v.5.0
No related merge requests found
......@@ -5,7 +5,7 @@
<groupId>edu.illinois.cs.cogcomp</groupId>
<artifactId>illinois-srl</artifactId>
<packaging>jar</packaging>
<version>5.1.1</version>
<version>5.1.2</version>
<url>http://cogcomp.cs.illinois.edu</url>
......
......@@ -12,6 +12,7 @@ public class FrameData {
private static class SenseFrameData {
Map<String, ArgumentData> argDescription = new HashMap<String, ArgumentData>();
String verbClass = "UNKNOWN";
String senseName;
List<Example> examples = new ArrayList<Example>();
}
......@@ -48,9 +49,10 @@ public class FrameData {
return lemma;
}
public void addSense(String sense, String verbClass) {
public void addSense(String sense, String senseName, String verbClass) {
this.senseFrameData.put(sense, new SenseFrameData());
this.senseFrameData.get(sense).verbClass = verbClass;
this.senseFrameData.get(sense).senseName = senseName;
}
public Set<String> getSenses() {
......@@ -92,4 +94,10 @@ public class FrameData {
l.addAll(this.getArgsForSense(s));
return l;
}
public String getSenseName(String sense) {
assert this.senseFrameData.containsKey(sense) : sense
+ " missing for predicate lemma " + this.lemma;
return this.senseFrameData.get(sense).senseName;
}
}
......@@ -139,6 +139,8 @@ public class FramesManager {
String sense = roleSet.getAttribute("id");
String senseName = roleSet.getAttribute("name");
// WTF frame makers?
if (sense.equals("lionise.01"))
sense = "lionize.01";
......@@ -158,7 +160,7 @@ public class FramesManager {
sense = sense.replaceAll(lemma + ".", "");
fData.addSense(sense, verbClass);
fData.addSense(sense, senseName, verbClass);
NodeList roles = roleSet.getElementsByTagName("role");
......
0% Loading or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment