Small tweak to ensemble textcat model
@@ -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:
