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

Question

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

在 Jupyter 笔记本中

如您所见，它根据训练好的模型正确预测

在 PyCharm 中

有谁知道为什么会这样？

标签： machine-learningscikit-learnpycharmdecision-treejoblib

Answer 1

您调用该joblib.load()方法，但不将输出分配给任何变量。在下一步中，您将尝试使用model上面任何地方都没有定义的变量，因此会出现错误。

你需要做：

model = joblib.load(...)