some new features
This commit is contained in:
@ -0,0 +1,42 @@
|
||||
"""Helper files for pickling"""
|
||||
from statsmodels.iolib.openfile import get_file_obj
|
||||
|
||||
|
||||
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)
|
||||
|
||||
|
||||
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)
|
||||
Reference in New Issue
Block a user