Python教程

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") # 导入
这篇关于Python学习笔记：sklearn.externals.joblib导入报错cannot import name 'joblib'的文章就介绍到这儿，希望我们推荐的文章对大家有所帮助，也希望大家多多支持为之网！
您可能喜欢
栏目导航