Update spacy/pipeline/spancat.py

Co-authored-by: Adriane Boyd <adrianeboyd@gmail.com>
This commit is contained in:
kadarakos 2023-03-03 15:55:42 +01:00 committed by GitHub
parent b972328337
commit fded200128
No known key found for this signature in database
GPG Key ID: 4AEE18F83AFDEB23

View File

@ -316,10 +316,10 @@ class SpanCategorizer(TrainablePipe):
) -> None:
"""Initialize the multi-label or multi-class span categorizer.
The 'single_label' argument configures whether the component
should only produce one label per span (multiclass) or if it
can produce multiple labels per span (multilabel). In the
multilabel case the classification layer is expected to be
Logistic and Softmax in the multiclass case.
should only produce one label per span (multi-class) or if it
can produce multiple labels per span (multi-label). In the
multi-label case the classification layer is expected to be
Logistic and Softmax in the multi-class case.
vocab (Vocab): The shared vocabulary.
model (thinc.api.Model): The Thinc Model powering the pipeline component.
suggester (Callable[[Iterable[Doc], Optional[Ops]], Ragged]): A function that suggests spans.