我有一个训练有素的决策树模型文件music-recommender.joblib
。当我使用Jupyter笔记本电脑时,我能够成功加载此训练模型并能够进行预测。但是我在pycharm中使用了相同的代码,并且显示错误。
在Jupyter Notebook中
如您所见,它是基于经过训练的模型正确预测的
在PyCharm中
有人知道为什么会这样吗?
答案 0 :(得分:1)
您调用joblib.load()
方法,但不要将输出分配给任何变量。并且在下一步中,您尝试使用model
变量,该变量在上方没有定义,因此会出错。
您需要这样做:
model = joblib.load(...)