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

Fix to/from disk in tagger

Browse Source
This commit is contained in:
Matthew Honnibal 2017-06-01 04:55:49 -05:00
parent 4c97371051
commit 5eae3b9a1e
1 changed files with 4 additions and 4 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

8
spacy/pipeline.pyx
View File

@ -160,7 +160,7 @@ class TokenVectorEncoder(object):
def to_bytes(self, **exclude):
serialize = OrderedDict((
('model', lambda: util.model_to_bytes(self.model)),
('model', lambda: self.model.to_bytes()),
('vocab', lambda: self.vocab.to_bytes())
))
return util.to_bytes(serialize, exclude)
@ -177,7 +177,7 @@ class TokenVectorEncoder(object):
def to_disk(self, path, **exclude):
serialize = OrderedDict((
('model', lambda p: p.open('wb').write(util.model_to_bytes(self.model))),
('model', lambda p: p.open('wb').write(self.model.to_bytes())),
('vocab', lambda p: self.vocab.to_disk(p))
))
util.to_disk(path, serialize, exclude)
@ -298,7 +298,7 @@ class NeuralTagger(object):
def to_bytes(self, **exclude):
serialize = OrderedDict((
('model', lambda: util.model_to_bytes(self.model)),
('model', lambda: self.model.to_bytes()),
('vocab', lambda: self.vocab.to_bytes())
))
return util.to_bytes(serialize, exclude)
@ -318,7 +318,7 @@ class NeuralTagger(object):
def to_disk(self, path, **exclude):
serialize = {
'model': lambda p: p.open('wb').write(util.model_to_bytes(self.model)),
'model': lambda p: p.open('wb').write(self.model.to_bytes()),
'vocab': lambda p: self.vocab.to_disk(p)
}
util.to_disk(path, serialize, exclude)