Esempio n. 1
0
    def save(self, path):
        for index, step in enumerate(self._steps):
            step_path = path + str(index)
            step.save(step_path)

        classes = [step.__class__ for step in self._steps]
        dump_pickle(classes, path, "steps.pickle")
        shutil.copytree(self._steps[-1].output_path, path+"scores")
Esempio n. 2
0
 def save(self, path=None):
     if path:
         dump_pickle(self._model, path)
     else:
         dump_pickle(self._model, self._model_path)