From 26d5d637e31ed0a0a038e8a2223b3d0ce85ced7e Mon Sep 17 00:00:00 2001 From: Lj Miranda Date: Tue, 17 Jan 2023 10:34:21 +0800 Subject: [PATCH] Add suggester documentation in Exclusive_SpanCategorizer --- spacy/pipeline/spancat_exclusive.py | 3 +++ 1 file changed, 3 insertions(+) diff --git a/spacy/pipeline/spancat_exclusive.py b/spacy/pipeline/spancat_exclusive.py index d1c3fd18a..86b9875af 100644 --- a/spacy/pipeline/spancat_exclusive.py +++ b/spacy/pipeline/spancat_exclusive.py @@ -141,6 +141,9 @@ class Exclusive_SpanCategorizer(SpanCategorizer): """Initialize the exclusive span categorizer. 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. + Spans are returned as a ragged array with two integer columns, for the + start and end positions. name (str): The component instance name, used to add entries to the losses during training. spans_key (str): Key of the Doc.spans dict to save the spans under.