create a list of all coref mentions for the document

starcount · Nov 1, 2017 · cc4b6b6 · cc4b6b6
1 parent b74d6bb
commit cc4b6b6
Showing 1 changed file with 3 additions and 0 deletions.
diff --git a/src/edu/stanford/nlp/pipeline/MentionAnnotator.java b/src/edu/stanford/nlp/pipeline/MentionAnnotator.java
@@ -1,5 +1,6 @@
 package edu.stanford.nlp.pipeline;
 import java.io.IOException;
+import java.util.ArrayList;
 import java.util.Arrays;
 import java.util.Collections;
 import java.util.HashSet;
@@ -98,11 +99,13 @@ public void annotate(Annotation annotation) {
       corefProperties.setProperty("removeNestedMentions", "true");
     }
     List<List<Mention>> mentions = md.findMentions(annotation, dictionaries, corefProperties);
+    annotation.set(CorefCoreAnnotations.CorefMentionsAnnotation.class , new ArrayList<Mention>());
     int mentionIndex = 0;
     int currIndex = 0;
     for (CoreMap sentence : sentences) {
       List<Mention> mentionsForThisSentence = mentions.get(currIndex);
       sentence.set(CorefCoreAnnotations.CorefMentionsAnnotation.class, mentionsForThisSentence);
+      annotation.get(CorefCoreAnnotations.CorefMentionsAnnotation.class).addAll(mentionsForThisSentence);
       // increment to next list of mentions
       currIndex++;
       // assign latest mentionID