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

Attempt to use registry correctly

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

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

@ -143,6 +143,11 @@ def get_candidate_mentions(
return (begins, ends) return (begins, ends)
@registry.misc("spacy.CorefCandidateGenerator.v0")
def create_mention_generator() -> Callable:
return get_candidate_mentions
def select_non_crossing_spans( def select_non_crossing_spans(
idxs: List[int], starts: List[int], ends: List[int], limit: int idxs: List[int], starts: List[int], ends: List[int], limit: int
) -> List[int]: ) -> List[int]:

4
spacy/pipeline/coref.py
View File

@ -31,7 +31,9 @@ max_span_width = 20
mention_limit = 3900 mention_limit = 3900
dropout = 0.3 dropout = 0.3
hidden = 1000 hidden = 1000
@get_mentions = "spacy.CorefCandidateGenerator.v0"
[model.get_mentions]
@misc = "spacy.CorefCandidateGenerator.v0"
[model.tok2vec] [model.tok2vec]
@architectures = "spacy.Tok2Vec.v2" @architectures = "spacy.Tok2Vec.v2"