Rooba
/
spaCy
mirror of https://github.com/explosion/spaCy.git
1
1
Fork 0
Code Issues Packages Projects Releases Wiki Activity

Fix to/from disk in tagger

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): def to_bytes(self, **exclude):
serialize = OrderedDict(( serialize = OrderedDict((
('model', lambda: util.model_to_bytes(self.model)), ('model', lambda: self.model.to_bytes()),
('vocab', lambda: self.vocab.to_bytes()) ('vocab', lambda: self.vocab.to_bytes())
)) ))
return util.to_bytes(serialize, exclude) return util.to_bytes(serialize, exclude)
@ -177,7 +177,7 @@ class TokenVectorEncoder(object):
def to_disk(self, path, **exclude): def to_disk(self, path, **exclude):
serialize = OrderedDict(( 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)) ('vocab', lambda p: self.vocab.to_disk(p))
)) ))
util.to_disk(path, serialize, exclude) util.to_disk(path, serialize, exclude)
@ -298,7 +298,7 @@ class NeuralTagger(object):
def to_bytes(self, **exclude): def to_bytes(self, **exclude):
serialize = OrderedDict(( serialize = OrderedDict((
('model', lambda: util.model_to_bytes(self.model)), ('model', lambda: self.model.to_bytes()),
('vocab', lambda: self.vocab.to_bytes()) ('vocab', lambda: self.vocab.to_bytes())
)) ))
return util.to_bytes(serialize, exclude) return util.to_bytes(serialize, exclude)
@ -318,7 +318,7 @@ class NeuralTagger(object):
def to_disk(self, path, **exclude): def to_disk(self, path, **exclude):
serialize = { 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) 'vocab': lambda p: self.vocab.to_disk(p)
} }
util.to_disk(path, serialize, exclude) util.to_disk(path, serialize, exclude)