explosion/spaCy
1
1
Fork 0
mirror of https://github.com/explosion/spaCy.git synced 2024-12-24 17:06:29 +03:00
Code Issues Packages Projects Releases Wiki Activity

Add __reduce__ methods on parser subclasses. Fixes pickling.

Browse Source
This commit is contained in:
Matthew Honnibal 2017-05-27 15:46:06 -05:00
parent 7cc9c3e9a6
commit 34bbad8e0e
1 changed files with 7 additions and 0 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

7
spacy/pipeline.pyx
View File

@ -335,6 +335,9 @@ cdef class NeuralDependencyParser(NeuralParser):
name = 'parser'
TransitionSystem = ArcEager
def __reduce__(self):
return (NeuralDependencyParser, (self.vocab, self.moves, self.model), None, None)
cdef class NeuralEntityRecognizer(NeuralParser):
name = 'entity'
@ -342,6 +345,10 @@ cdef class NeuralEntityRecognizer(NeuralParser):
nr_feature = 6
def __reduce__(self):
return (NeuralEntityRecognizer, (self.vocab, self.moves, self.model), None, None)
cdef class BeamDependencyParser(BeamParser):
TransitionSystem = ArcEager