From f436efd8a4dfbd0b8de5c6aa7b368f2259e06b98 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sat, 23 Mar 2019 16:46:30 +0100 Subject: [PATCH] Small tweak to ensemble textcat model --- spacy/_ml.py | 2 +- 1 file changed, 1 insertion(+), 1 deletion(-) diff --git a/spacy/_ml.py b/spacy/_ml.py index ad061d79c..2de876ec9 100644 --- a/spacy/_ml.py +++ b/spacy/_ml.py @@ -592,7 +592,7 @@ def build_text_classifier(nr_class, width=64, **cfg): ) linear_model = build_bow_text_classifier( - nr_class, ngram_size=cfg.get("ngram_size", 1), no_output_layer=True) + nr_class, ngram_size=cfg.get("ngram_size", 1), exclusive_classes=False) if cfg.get('exclusive_classes'): output_layer = Softmax(nr_class, nr_class * 2) else: