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