Fix exclusive_classes in textcat ensemble (#5166)

Pass the exclusive_classes setting to the bow model within the ensemble textcat model.
explosion · Mar 29, 2020 · d47b810 · d47b810
1 parent e904958
commit d47b810
Showing 1 changed file with 4 additions and 2 deletions.
diff --git a/spacy/_ml.py b/spacy/_ml.py
@@ -693,9 +693,11 @@ def build_text_classifier(nr_class, width=64, **cfg):
         )
 
         linear_model = build_bow_text_classifier(
-            nr_class, ngram_size=cfg.get("ngram_size", 1), exclusive_classes=False
+            nr_class,
+            ngram_size=cfg.get("ngram_size", 1),
+            exclusive_classes=cfg.get("exclusive_classes", False),
         )
-        if cfg.get("exclusive_classes"):
+        if cfg.get("exclusive_classes", False):
             output_layer = Softmax(nr_class, nr_class * 2)
         else:
             output_layer = (