From 5260268f70885c1b3ec6581b7184cf0b29d03d04 Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Sun, 29 Apr 2018 15:48:53 +0200 Subject: [PATCH] Fix textcat after merge --- spacy/pipeline.pyx | 7 +++---- 1 file changed, 3 insertions(+), 4 deletions(-) diff --git a/spacy/pipeline.pyx b/spacy/pipeline.pyx index 2a21bebb0..cb838b449 100644 --- a/spacy/pipeline.pyx +++ b/spacy/pipeline.pyx @@ -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()