Fix default argument in TextCategorizer.Model (resolves #3221)

This commit is contained in:
Ines Montani 2019-02-05 12:33:47 +01:00
parent 89ad095900
commit 5e16490d9d

View File

@ -1145,7 +1145,7 @@ class TextCategorizer(Pipe):
name = 'textcat'
@classmethod
def Model(cls, nr_class, **cfg):
def Model(cls, nr_class=1, **cfg):
embed_size = util.env_opt("embed_size", 2000)
if "token_vector_width" in cfg:
token_vector_width = cfg["token_vector_width"]