Add custom entities specific to my research around concepts and entit…

…ies in academic papers.
aoldoni · Feb 22, 2017 · ae67822 · ae67822
1 parent 80cbe15
commit ae67822
Show file tree

Hide file tree

Showing 2 changed files with 9 additions and 1 deletion.
diff --git a/edu/stanford/nlp/ie/machinereading/domains/roth/RothCONLL04Reader.java b/edu/stanford/nlp/ie/machinereading/domains/roth/RothCONLL04Reader.java
@@ -70,6 +70,12 @@ else if(ner.equalsIgnoreCase("Org"))
       return "ORGANIZATION";
     else if(ner.equalsIgnoreCase("Other"))
       return "OTHER";
+    else if(ner.equalsIgnoreCase("Concept"))
+      return "CONCEPT";
+    else if(ner.equalsIgnoreCase("Entity"))
+      return "ENTITY";
+    else if(ner.equalsIgnoreCase("Paper"))
+      return "PAPER";
     throw new RuntimeException("Cannot normalize ner tag " + ner);
   }
 

diff --git a/edu/stanford/nlp/ie/machinereading/domains/roth/RothEntityExtractor.java b/edu/stanford/nlp/ie/machinereading/domains/roth/RothEntityExtractor.java
@@ -22,7 +22,9 @@ public RothEntityExtractor() {
     entityTagForNer.put("person", "PEOPLE");
     entityTagForNer.put("organization", "ORGANIZATION");
     entityTagForNer.put("location", "LOCATION");
-
+    entityTagForNer.put("concept", "CONCEPT");
+    entityTagForNer.put("entity", "ENTITY");
+    entityTagForNer.put("paper", "PAPER");
   }
 
   @Override