explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2025-03-03 19:08:06 +03:00
Code Issues Packages Projects Releases Wiki Activity

Fix transition system to/from disk

Browse Source
This commit is contained in:
Matthew Honnibal 2017-05-31 13:44:00 +02:00
parent b1469d3360
commit 097ab9c6e4
1 changed files with 5 additions and 14 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

19
spacy/syntax/transition_system.pyx
View File

@ -157,22 +157,13 @@ cdef class TransitionSystem:
return 1 return 1
def to_disk(self, path, **exclude): def to_disk(self, path, **exclude):
actions = list(self.move_names) with path.open('wb') as file_:
deserializers = { file_.write(self.to_bytes(**exclude))
'actions': lambda p: ujson.dump(p.open('w'), actions),
'strings': lambda p: self.strings.to_disk(p)
}
util.to_disk(path, deserializers, exclude)
def from_disk(self, path, **exclude): def from_disk(self, path, **exclude):
actions = [] with path.open('rb') as file_:
deserializers = { byte_data = file_.read()
'strings': lambda p: self.strings.from_disk(p), self.from_bytes(byte_data, **exclude)
'actions': lambda p: actions.extend(ujson.load(p.open()))
}
util.from_disk(path, deserializers, exclude)
for move, label in actions:
self.add_action(move, label)
return self return self
def to_bytes(self, **exclude): def to_bytes(self, **exclude):