From c2357d3ba075f0733505ca69ac114705872bb07b Mon Sep 17 00:00:00 2001 From: Matthew Honnibal Date: Tue, 25 Sep 2018 10:58:13 +0200 Subject: [PATCH] Fix morphologizer class --- spacy/_morphologizer.pyx | 13 ++++++++++--- 1 file changed, 10 insertions(+), 3 deletions(-) diff --git a/spacy/_morphologizer.pyx b/spacy/_morphologizer.pyx index ca857296e..2fa092faa 100644 --- a/spacy/_morphologizer.pyx +++ b/spacy/_morphologizer.pyx @@ -12,7 +12,7 @@ from thinc.api import chain from thinc.neural.util import to_categorical, copy_array from . import util from .pipe import Pipe -from ._ml import Tok2Vec, build_tagger_model +from ._ml import Tok2Vec, build_morphologizer_model from ._ml import link_vectors_to_models, zero_init, flatten from ._ml import create_default_optimizer from .errors import Errors, TempErrors @@ -20,6 +20,7 @@ from .compat import json_dumps, basestring_ from .tokens.doc cimport Doc from .vocab cimport Vocab from .morphology cimport Morphology +from .pipeline import Pipe class Morphologizer(Pipe): @@ -50,7 +51,7 @@ class Morphologizer(Pipe): def __call__(self, doc): features, tokvecs = self.predict([doc]) - self.set_annotations([doc], tags, tensors=tokvecs) + self.set_annotations([doc], features, tensors=tokvecs) return doc def pipe(self, stream, batch_size=128, n_threads=-1): @@ -81,7 +82,7 @@ class Morphologizer(Pipe): cdef Doc doc cdef Vocab vocab = self.vocab for i, doc in enumerate(docs): - doc_feat_ids = batch_feat_ids[i] + doc_feat_ids = batch_feature_ids[i] if hasattr(doc_feat_ids, 'get'): doc_feat_ids = doc_feat_ids.get() # Convert the neuron indices into feature IDs. @@ -129,3 +130,9 @@ class Morphologizer(Pipe): def use_params(self, params): with self.model.use_params(params): yield + +def scores_to_guesses(scores, out_sizes): + raise NotImplementedError + +def feature_to_column(feature): + raise NotImplementedError