为什么pycharm在加载.joblib文件时出错

时间:2019-05-23 06:30:17

标签: machine-learning scikit-learn pycharm decision-tree joblib

我有一个训练有素的决策树模型文件music-recommender.joblib。当我使用Jupyter笔记本电脑时,我能够成功加载此训练模型并能够进行预测。但是我在pycharm中使用了相同的代码,并且显示错误。

在Jupyter Notebook中

enter image description here

如您所见,它是基于经过训练的模型正确预测的

在PyCharm中

enter image description here

有人知道为什么会这样吗?

1 个答案:

答案 0 :(得分:1)

您调用joblib.load()方法,但不要将输出分配给任何变量。并且在下一步中,您尝试使用model变量,该变量在上方没有定义,因此会出错。

您需要这样做:

model = joblib.load(...)