explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-09-21 11:29:13 +03:00
Code Issues Packages Projects Releases Wiki Activity

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

Browse Source
This commit is contained in:
Ines Montani 2019-02-05 12:33:47 +01:00
parent 89ad095900
commit 5e16490d9d
1 changed files with 1 additions and 1 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

2
spacy/pipeline.pyx
View File

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