def save_info(self) -> dict:
info = self.get_info()
save_pkl.save(path=self.path + self.model_info_name, object=info)
json_path = self.path + self.model_info_json_name
save_json.save(path=json_path, obj=info)
return info
def save_info(self, include_model_info=False):
info = self.get_info(include_model_info=include_model_info)
save_pkl.save(path=self.path + self.learner_info_name, object=info)
save_json.save(path=self.path + self.learner_info_json_name, obj=info)
return info
