之前跑完模型保存都是通过以下脚本保存。
# 保存模型
from sklearn.externals import joblib
joblib.dump(model_lgb, r"E:\model.pkl") # 导出
# model_lgb = joblib.load(r"E:\model.pkl") # 导入
最近升级 sklearn 0.24
包之后发现这种保存方式会报错。
scikit-learn 0.24.2 pypi_0 pypi
Traceback (most recent call last):
File "<ipython-input-21-02f086d2bd8e>", line 1, in <module>
from sklearn.externals import joblib
ImportError: cannot import name 'joblib'
遂查找一番,才知道 sklearn
包在0.23版本后取消了 joblib
模块,如果要使用,直接安装 joblib
包即可。
# 安装
pip install joblib # 更新sklearn时会依赖安装
# 保存模型
import joblib
joblib.dump(model_lgb, r"E:\model.pkl") # 导出
# model_lgb = joblib.load(r"E:\model.pkl") # 导入