Python学习笔记：sklearn.externals.joblib导入报错cannot import name 'joblib'

之前跑完模型保存都是通过以下脚本保存。

# 保存模型
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") # 导入