explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-01-30 19:24:07 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix transition has_gold

Browse Source
This commit is contained in:
Matthew Honnibal 2021-01-25 16:48:45 +11:00
parent b2044d510e
commit bd04ea0b02
2 changed files with 4 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline/_parser_internals/arc_eager.pyx
View File

@ -757,6 +757,8 @@ cdef class ArcEager(TransitionSystem):
return list(arcs) return list(arcs)
def has_gold(self, Example eg, start=0, end=None): def has_gold(self, Example eg, start=0, end=None):
if end is not None and end < 0:
end = None
for word in eg.y[start:end]: for word in eg.y[start:end]:
if word.dep != 0: if word.dep != 0:
return True return True

2
spacy/pipeline/_parser_internals/ner.pyx
View File

@ -266,6 +266,8 @@ cdef class BiluoPushDown(TransitionSystem):
return BiluoGold(self, state, example) return BiluoGold(self, state, example)
def has_gold(self, Example eg, start=0, end=None): def has_gold(self, Example eg, start=0, end=None):
if end is not None and end < 0:
end = None
for word in eg.y[start:end]: for word in eg.y[start:end]:
if word.ent_iob != 0: if word.ent_iob != 0:
return True return True