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

black

Browse Source
This commit is contained in:
kadarakos 2023-01-31 16:30:12 +00:00
parent dceeb02b94
commit 8a807ef1dd
1 changed files with 8 additions and 6 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

14
spacy/pipeline/spancat.py
View File

@ -71,7 +71,9 @@ depth = 4
"""
DEFAULT_SPANCAT_MODEL = Config().from_str(spancat_default_config)["model"]
DEFAULT_SPANCAT_SINGLELABEL_MODEL = Config().from_str(spancat_singlelabel_default_config)["model"]
DEFAULT_SPANCAT_SINGLELABEL_MODEL = Config().from_str(
spancat_singlelabel_default_config
)["model"]
@runtime_checkable
@ -191,7 +193,7 @@ def make_spancat(
"negative_weight": 1.0,
"suggester": {"@misc": "spacy.ngram_suggester.v1", "sizes": [1, 2, 3]},
"scorer": {"@scorers": "spacy.spancat_scorer.v1"},
"allow_overlap": True
"allow_overlap": True,
},
default_score_weights={"spans_sc_f": 1.0, "spans_sc_p": 0.0, "spans_sc_r": 0.0},
)
@ -237,7 +239,7 @@ def make_spancat_singlelabel(
allow_overlap=allow_overlap,
name=name,
scorer=scorer,
single_label=True
single_label=True,
)
@ -332,7 +334,7 @@ class SpanCategorizer(TrainablePipe):
"threshold": threshold,
"max_positive": max_positive,
"negative_weight": negative_weight,
"allow_overlap": allow_overlap
"allow_overlap": allow_overlap,
}
self.vocab = vocab
self.suggester = suggester
@ -471,7 +473,7 @@ class SpanCategorizer(TrainablePipe):
indices_i,
scores[offset : offset + indices.lengths[i]],
labels, # type: ignore[arg-type]
allow_overlap
allow_overlap,
)
else:
doc.spans[self.key] = self._make_span_group_multilabel(
@ -638,7 +640,7 @@ class SpanCategorizer(TrainablePipe):
indices: Ints2d,
scores: Floats2d,
labels: List[str],
allow_overlap: bool = True
allow_overlap: bool = True,
) -> SpanGroup:
spans = SpanGroup(doc, name=self.key)
max_positive = self.cfg["max_positive"]