INNER CODE UNIT · Python
_unload_params
CamDavidsonPilon/lifetimes · lifetimes/fitters/__init__.py:34
def _unload_params(self, *args):
if not hasattr(self, "params_"):
raise ValueError("Model has not been fit yet. Please call the .fit" " method first.")
return [self.params_[x] for x in args]
def save_model(self, path, save_data=True, save_generate_data_method=True, values_to_save=None):
"""
Save model with dill package.
Parameters
----------
path: str
Path where to save model.
save_data: bool, optional
Whether to save data from fitter.data to pickle object
save_generate_data_method: bool, optional
Whether to save generate_new_data method (if it exists) from
fitter.generate_new_data to pickle object.