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