def save(self, out_dir): """Dispatcher function for saving.""" params = { "model_params": self.model_params, "task_types": self.task_types, "model_class": self.__class__ } save_to_disk(params, Model.get_params_filename(out_dir))
def save(self, out_dir): """Saves sklearn model to disk using joblib.""" super(SklearnModel, self).save(out_dir) save_to_disk(self.raw_model, self.get_model_filename(out_dir))
def save(self, out_dir): """Dispatcher function for saving.""" params = {"model_params" : self.model_params, "task_types" : self.task_types, "model_type": self.model_type} save_to_disk(params, Model.get_params_filename(out_dir))