Update spacy/pipeline/spancat.py

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

View File

@ -322,6 +322,7 @@ class SpanCategorizer(TrainablePipe):
can produce multiple labels per span (multi-label). In the can produce multiple labels per span (multi-label). In the
multi-label case the classification layer is expected to be multi-label case the classification layer is expected to be
Logistic and Softmax in the multi-class 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.