explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-04-29 21:33:42 +03:00
Code Issues Packages Projects Releases Wiki Activity

Update how spancat_exclusive is constructed

Browse Source 
In this commit, I added the following:
- Put the default values of negative_weight and allow_overlap
    in the default_config dictionary.
- Rename make_spancat -> make_exclusive_spancat
This commit is contained in:
Lj Miranda 2023-01-17 10:17:29 +08:00
parent eac09df704
commit e61f0a4035
1 changed files with 7 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

12
spacy/pipeline/spancat_exclusive.py
View File

@ -46,22 +46,24 @@ DEFAULT_EXCL_SPANCAT_MODEL = Config().from_str(spancat_excl_default_config)["mod
"spancat_exclusive", "spancat_exclusive",
assigns=["doc.spans"], assigns=["doc.spans"],
default_config={ default_config={
"spans_key": "sc",
"model": DEFAULT_EXCL_SPANCAT_MODEL,
"suggester": {"@misc": "spacy.ngram_suggester.v1", "sizes": [1, 2, 3]}, "suggester": {"@misc": "spacy.ngram_suggester.v1", "sizes": [1, 2, 3]},
"model": DEFAULT_EXCL_SPANCAT_MODEL,
"spans_key": "sc",
"scorer": {"@scorers": "spacy.spancat_scorer.v1"}, "scorer": {"@scorers": "spacy.spancat_scorer.v1"},
"negative_weight": 1.0,
"allow_overlap": True,
}, },
default_score_weights={"spans_sc_f": 1.0, "spans_sc_p": 0.0, "spans_sc_r": 0.0}, default_score_weights={"spans_sc_f": 1.0, "spans_sc_p": 0.0, "spans_sc_r": 0.0},
) )
def make_spancat( def make_exclusive_spancat(
nlp: Language, nlp: Language,
name: str, name: str,
suggester: Suggester, suggester: Suggester,
model: Model[Tuple[List[Doc], Ragged], Floats2d], model: Model[Tuple[List[Doc], Ragged], Floats2d],
spans_key: str, spans_key: str,
scorer: Optional[Callable], scorer: Optional[Callable],
negative_weight: float = 1.0, negative_weight: float,
allow_overlap: bool = True, allow_overlap: bool,
) -> "Exclusive_SpanCategorizer": ) -> "Exclusive_SpanCategorizer":
"""Create an Exclusive_SpanCategorizer component. The span categorizer consists of two """Create an Exclusive_SpanCategorizer component. The span categorizer consists of two
parts: a suggester function that proposes candidate spans, and a labeler parts: a suggester function that proposes candidate spans, and a labeler