INNER CODE UNIT · Python
save_model
CamDavidsonPilon/lifetimes · lifetimes/fitters/__init__.py:39
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.
values_to_save: list, optional
Placeholders for original attributes for saving object. If None
will be extended to attr_list length like [None] * len(attr_list)
"""