Example #1
0
 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))
Example #2
0
 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))
Example #3
0
 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))
Example #4
0
 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))