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

Make get_sentence_map work with init

Browse Source 
When sentences are not available, just treat the whole doc as one
sentence. A reasonable general fallback, but important due to the init
call, where upstream components aren't run.
This commit is contained in:
Paul O'Leary McCann 2021-05-18 19:54:54 +09:00
parent 883c137b26
commit a7d9c8156d
2 changed files with 15 additions and 9 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

1
spacy/ml/models/coref.py
View File

@ -145,6 +145,7 @@ def span_embeddings_forward(
tokvecs, docs = inputs tokvecs, docs = inputs
#TODO fix this
dim = tokvecs[0].shape[1] dim = tokvecs[0].shape[1]
get_mentions = model.attrs["get_mentions"] get_mentions = model.attrs["get_mentions"]

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

@ -1,6 +1,6 @@
from thinc.types import Ints2d from thinc.types import Ints2d
from spacy.tokens import Doc from spacy.tokens import Doc
from typing import List, Tuple, Callable from typing import List, Tuple, Callable, Any
from ...util import registry from ...util import registry
# type alias to make writing this less tedious # type alias to make writing this less tedious
@ -109,6 +109,7 @@ def get_predicted_clusters(
def get_sentence_map(doc: Doc): def get_sentence_map(doc: Doc):
"""For the given span, return a list of sentence indexes.""" """For the given span, return a list of sentence indexes."""
try:
si = 0 si = 0
out = [] out = []
for sent in doc.sents: for sent in doc.sents:
@ -116,6 +117,10 @@ def get_sentence_map(doc: Doc):
out.append(si) out.append(si)
si += 1 si += 1
return out return out
except ValueError:
# If there are no sents then just return dummy values.
# Shouldn't happen in general training, but typical in init.
return [0] * len(doc)
def get_candidate_mentions( def get_candidate_mentions(
@ -144,7 +149,7 @@ def get_candidate_mentions(
@registry.misc("spacy.CorefCandidateGenerator.v0") @registry.misc("spacy.CorefCandidateGenerator.v0")
def create_mention_generator() -> Callable: def create_mention_generator() -> Any:
return get_candidate_mentions return get_candidate_mentions