编程语言
首页 > 编程语言> > Python学习笔记:sklearn.externals.joblib导入报错cannot import name 'joblib'

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") # 导入

标签:lgb,cannot,报错,joblib,import,model,pkl,sklearn
来源: https://www.cnblogs.com/hider/p/15075259.html