Commit
·
2a150f0
1
Parent(s):
90d5d25
algorithm improved
Browse files
ud.py
CHANGED
|
@@ -38,6 +38,9 @@ class BellmanFordTokenClassificationPipeline(TokenClassificationPipeline):
|
|
| 38 |
if p.startswith("I-"):
|
| 39 |
w[i-1]["score"]=min(w[i-1]["score"],t["score"])
|
| 40 |
w[i-1]["end"]=w.pop(i)["end"]
|
|
|
|
|
|
|
|
|
|
| 41 |
elif p.startswith("B-"):
|
| 42 |
t["entity_group"]=p[2:]
|
| 43 |
else:
|
|
|
|
| 38 |
if p.startswith("I-"):
|
| 39 |
w[i-1]["score"]=min(w[i-1]["score"],t["score"])
|
| 40 |
w[i-1]["end"]=w.pop(i)["end"]
|
| 41 |
+
elif i>0 and w[i-1]["end"]>t["start"]:
|
| 42 |
+
w[i-1]["score"]=min(w[i-1]["score"],t["score"])
|
| 43 |
+
w[i-1]["end"]=w.pop(i)["end"]
|
| 44 |
elif p.startswith("B-"):
|
| 45 |
t["entity_group"]=p[2:]
|
| 46 |
else:
|