Source code for statsmodels.iolib.smpickle
"""Helper files for pickling"""
from statsmodels.iolib.openfile import get_file_obj
[docs]
def save_pickle(obj, fname):
"""
Save the object to file via pickling.
Parameters
----------
fname : {str, pathlib.Path}
Filename to pickle to
"""
import pickle
with get_file_obj(fname, "wb") as fout:
pickle.dump(obj, fout, protocol=-1)
[docs]
def load_pickle(fname):
"""
Load a previously saved object
.. warning::
Loading pickled models is not secure against erroneous or maliciously
constructed data. Never unpickle data received from an untrusted or
unauthenticated source.
Parameters
----------
fname : {str, pathlib.Path}
Filename to unpickle
Notes
-----
This method can be used to load *both* models and results.
"""
import pickle
with get_file_obj(fname, "rb") as fin:
return pickle.load(fin)
Last update:
Oct 12, 2024