mirror of
https://github.com/explosion/spaCy.git
synced 2025-07-11 00:32:40 +03:00
black
This commit is contained in:
parent
dceeb02b94
commit
8a807ef1dd
|
@ -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"]
|
||||
|
|
Loading…
Reference in New Issue
Block a user