explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-03-28 13:54:12 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix textcat after merge

Browse Source
This commit is contained in:
Matthew Honnibal 2018-04-29 15:48:53 +02:00
parent ad3d56c3ba
commit 5260268f70
1 changed files with 3 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/pipeline.pyx
View File

@ -888,8 +888,8 @@ class TextCategorizer(Pipe):
name = 'textcat'
@classmethod
def Model(cls, **cfg):
return build_text_classifier(**cfg)
def Model(cls, nr_class, **cfg):
return build_text_classifier(nr_class, **cfg)
def __init__(self, vocab, model=True, **cfg):
self.vocab = vocab
@ -981,8 +981,7 @@ class TextCategorizer(Pipe):
if self.model is True:
self.cfg['pretrained_vectors'] = kwargs.get('pretrained_vectors')
self.model = self.Model(len(self.labels), token_vector_width,
**self.cfg)
self.model = self.Model(len(self.labels), **self.cfg)
link_vectors_to_models(self.vocab)
if sgd is None:
sgd = self.create_optimizer()