彌補srl的資訊漏失 HIT-SCIR#699

kanasimi · Jun 1, 2024 · f8a31cf · f8a31cf
1 parent 17f689c
commit f8a31cf
Showing 1 changed file with 3 additions and 3 deletions.
diff --git a/python/interface/ltp/nerual.py b/python/interface/ltp/nerual.py
@@ -196,7 +196,7 @@ def pipeline(
                         words = sentences[idx]
                         new_store.append(
                             [
-                                (tag, "".join(words[start : end + 1]))
+                                (tag, "".join(words[start : end + 1]), start, end)
                                 for tag, start, end in get_entities(sent)
                             ]
                         )
@@ -209,12 +209,12 @@ def pipeline(
 
                         for item, predicate in enumerate(words):
                             arguments = [
-                                (tag, "".join(words[start : end + 1]))
+                                (tag, "".join(words[start : end + 1]), start, end)
                                 for tag, start, end in get_entities(sent[item])
                             ]
                             if arguments:
                                 new_store[-1].append(
-                                    {"predicate": predicate, "arguments": arguments}
+                                    {"index": item, "predicate": predicate, "arguments": arguments}
                                 )
                     store[task] = new_store