From 51a53de2399b63d0e7a72d1bea441a49c70e005e Mon Sep 17 00:00:00 2001 From: kadarakos Date: Mon, 6 Mar 2023 15:14:35 +0100 Subject: [PATCH 1/2] Update spacy/pipeline/spancat.py Co-authored-by: Adriane Boyd --- spacy/pipeline/spancat.py | 2 -- 1 file changed, 2 deletions(-) diff --git a/spacy/pipeline/spancat.py b/spacy/pipeline/spancat.py index 189c5ae41..a0addca2d 100644 --- a/spacy/pipeline/spancat.py +++ b/spacy/pipeline/spancat.py @@ -306,8 +306,6 @@ class SpanCategorizer(TrainablePipe): vocab: Vocab, model: Model[Tuple[List[Doc], Ragged], Floats2d], suggester: Suggester, - # TODO Not sure what's the best default name when it can both be spancat - # and spancat_singlelabel name: str = "spancat", *, add_negative_label: bool = False, From 308002f00d490bce5ba88aadb48f80fe9f7f7db5 Mon Sep 17 00:00:00 2001 From: kadarakos Date: Mon, 6 Mar 2023 15:14:55 +0100 Subject: [PATCH 2/2] Update spacy/pipeline/spancat.py Co-authored-by: Adriane Boyd --- spacy/pipeline/spancat.py | 1 + 1 file changed, 1 insertion(+) diff --git a/spacy/pipeline/spancat.py b/spacy/pipeline/spancat.py index a0addca2d..d8c187c0a 100644 --- a/spacy/pipeline/spancat.py +++ b/spacy/pipeline/spancat.py @@ -322,6 +322,7 @@ class SpanCategorizer(TrainablePipe): 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.