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

don't rely on default arguments

Browse Source
This commit is contained in:
kadarakos 2023-02-02 10:36:36 +00:00
parent 5ccb154972
commit d13e494abd
1 changed files with 20 additions and 10 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

30
spacy/pipeline/spancat.py
View File

@ -176,13 +176,16 @@ def make_spancat(
""" """
return SpanCategorizer( return SpanCategorizer(
nlp.vocab, nlp.vocab,
suggester=suggester,
model=model, model=model,
spans_key=spans_key, suggester=suggester,
threshold=threshold,
max_positive=max_positive,
name=name, name=name,
scorer=scorer, single_label=False,
spans_key=spans_key,
negative_weight=None,
allow_overlap=None,
max_positive=max_positive,
threshold=threshold,
scorer=scorer
) )
@ -236,14 +239,16 @@ def make_spancat_singlelabel(
""" """
return SpanCategorizer( return SpanCategorizer(
nlp.vocab, nlp.vocab,
suggester=suggester,
model=model, model=model,
suggester=suggester,
name=name,
single_label=True,
spans_key=spans_key, spans_key=spans_key,
negative_weight=negative_weight, negative_weight=negative_weight,
allow_overlap=allow_overlap, allow_overlap=allow_overlap,
name=name, max_positive=None,
scorer=scorer, threshold=None,
single_label=True, scorer=scorer
) )
@ -488,7 +493,12 @@ class SpanCategorizer(TrainablePipe):
for i, doc in enumerate(docs): for i, doc in enumerate(docs):
indices_i = indices[i].dataXd indices_i = indices[i].dataXd
if self.single_label: if self.single_label:
allow_overlap = cast(bool, self.cfg["allow_overlap"]) allow_overlap = self.cfg["allow_overlap"]
# Interpret None as False if allow_overlap is not provided
if allow_overlap is None:
allow_overlap = False
else:
allow_overlap = cast(bool, self.cfg["allow_overlap"])
doc.spans[self.key] = self._make_span_group_singlelabel( doc.spans[self.key] = self._make_span_group_singlelabel(
doc, doc,
indices_i, indices_i,