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")
def save(self, path=None): if path: dump_pickle(self._model, path) else: dump_pickle(self._model, self._model_path)