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

Minor optimization

Browse Source
This commit is contained in:
Paul O'Leary McCann 2021-06-17 21:10:46 +09:00
parent cb2364cf83
commit fce804a79f
1 changed files with 6 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

9
spacy/ml/models/coref_util.py
View File

@ -143,9 +143,12 @@ def get_candidate_mentions(
si = sentence_map[tok.i] # sentence index
for ii in range(1, max_span_width):
ei = tok.i + ii # end index
if ei < len(doc) and sentence_map[ei] == si:
begins.append(tok.i)
ends.append(ei)
if ei > len(doc) or sentence_map[ei] != si:
continue
begins.append(tok.i)
ends.append(ei)
return (begins, ends)