diff --git a/spacy/pipeline/textcat.py b/spacy/pipeline/textcat.py index 76726d436..0378802ac 100644 --- a/spacy/pipeline/textcat.py +++ b/spacy/pipeline/textcat.py @@ -405,4 +405,4 @@ class TextCategorizer(TrainablePipe): raise ValueError(Errors.E895.format(value=ex.reference.cats)) for val in ex.reference.cats.values(): if not (val == 1.0 or val == 0.0): - raise ValueError(Errors.E852.format(val=val)) + raise ValueError(Errors.E851.format(val=val)) diff --git a/spacy/pipeline/textcat_multilabel.py b/spacy/pipeline/textcat_multilabel.py index b777e3bec..78750dabc 100644 --- a/spacy/pipeline/textcat_multilabel.py +++ b/spacy/pipeline/textcat_multilabel.py @@ -197,7 +197,7 @@ class MultiLabel_TextCategorizer(TextCategorizer): for eg in subbatch: for val in eg.reference.cats.values(): if not (val == 1.0 or val == 0.0): - raise ValueError(Errors.E852.format(val=val)) + raise ValueError(Errors.E851.format(val=val)) doc_sample = [eg.reference for eg in subbatch] label_sample, _ = self._examples_to_truth(subbatch) self._require_labels()