explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-08-10 07:04:53 +03:00
Code Issues Packages Projects Releases Wiki Activity

Minor fixes

Browse Source
This commit is contained in:
Paul O'Leary McCann 2021-05-17 14:52:30 +09:00
parent 7c42a8c90a
commit 91b111467b
2 changed files with 4 additions and 3 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

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

@ -1,6 +1,7 @@
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 from typing import List, Tuple, Callable
from ...util import registry
# type alias to make writing this less tedious # type alias to make writing this less tedious
MentionClusters = List[List[Tuple[int, int]]] MentionClusters = List[List[Tuple[int, int]]]

4
spacy/pipeline/coref.py
View File

@ -73,7 +73,7 @@ def make_coref(
) -> "CoreferenceResolver": ) -> "CoreferenceResolver":
"""Create a CoreferenceResolver component.""" """Create a CoreferenceResolver component."""
return CoreferenceResolver(nlp.vocab, model, name, span_cluster_prefix) return CoreferenceResolver(nlp.vocab, model, name, span_cluster_prefix=span_cluster_prefix)
class CoreferenceResolver(TrainablePipe): class CoreferenceResolver(TrainablePipe):
@ -88,7 +88,7 @@ class CoreferenceResolver(TrainablePipe):
model: Model, model: Model,
name: str = "coref", name: str = "coref",
*, *,
span_mentions: str, span_mentions: str = "coref_mentions",
span_cluster_prefix: str, span_cluster_prefix: str,
) -> None: ) -> None:
"""Initialize a coreference resolution component. """Initialize a coreference resolution component.