无法在Python中加载ML模型

Question

我在Python中构建了一个模型，并使用joblib包中的sklearn.externals保存了该模型：

from sklearn.externals import joblib

joblib.dump(rf_Prob_F, 'Model.pkl') 

当我尝试使用以下命令调用模型时，会出现错误：

from sklearn.externals import joblib

rf_Prob_F = joblib.load(rf_Prob_F, 'Model.pkl')

  

NameError：名称'rf_Prob_F'未定义

我错过了什么？

感谢您的帮助！

Answer 1

正如documentation所述 对于joblib.load()，您只需要将文件名作为参数：

rf_Prob_F = joblib.load('Model.pkl')