【Python】将statsmodels库训练好的模型OLS保存本地

【参考】

1.https://ask.helplib.com/python/post_12850927

2.https://blog.youkuaiyun.com/qq_33363973/article/details/77881168

 

【代码】

1.因为文件的类型是字节bytes型的， 所以写入的模式必须是'wb'，读取是'rb'。

import statsmodels.api as sm
import pickle

# 训练模型
model = sm.OLS(y,X).fit()

# 保存本地
filepath = r'model/model.pkl'
with open(filepath, 'wb') as f:
    pickle.dump(model, f)

# 从本地导入使用
with open(filepath, 'rb') as f:
    load_model = pickle.load(f)