first draft of corefud.GuessSpan

martinpopel · martinpopel · commit bd0f24b75fcc · 2023-01-31T20:16:31.000+01:00
diff --git a/udapi/block/corefud/guessspan.py b/udapi/block/corefud/guessspan.py
@@ -0,0 +1,9 @@
+from udapi.core.block import Block
+
+class GuessSpan(Block):
+    """Block corefud.GuessSpan heuristically fills mention spans, while keeping mention.head"""
+
+    def process_coref_mention(self, mention):
+        mention.words = mention.head.descendants(add_self=True)
+        # TODO add empty nodes that are causing gaps
+        # TODO add heuristics from corefud.PrintMentions almost_forest=1
