mirror of
				https://github.com/explosion/spaCy.git
				synced 2025-10-31 16:07:41 +03:00 
			
		
		
		
	Fix morphologizer class
This commit is contained in:
		
							parent
							
								
									e6dde97295
								
							
						
					
					
						commit
						c2357d3ba0
					
				|  | @ -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 | ||||
|  |  | |||
		Loading…
	
		Reference in New Issue
	
	Block a user