def __getstate__(self):
'''This function is neccessary for pickling'''
# Translate everything but the svm because that cannot be simply pickled.
state = {}
for key,value in self.__dict__.iteritems():
if key == '_model':
filename = tempfile.mktemp()
self._model.save(filename)
buffer = open(filename).read()
os.remove(filename)
state[key] = buffer
continue
state[key] = value
return state
评论列表
文章目录