小编Joa*_*oao的帖子

Auto-sklearn:如何加载pickle文件并运行predict()

我已经使用 auto-sklearn 安装了一个分类模型,并设法将其保存到带有 pickle 的文件中。

x = automl.show_models()
results = {"ensemble": x}
pickle.dump(results, open('file.pickle','wb'))
Run Code Online (Sandbox Code Playgroud)

我还设法重新加载模型。

automl = pickle.load(open('file.pickle','rb'))
Run Code Online (Sandbox Code Playgroud)

但我无法设法使用重新加载的模型对新数据进行预测。当我跑步时:

y_hat = automl.predict(X_test)
Run Code Online (Sandbox Code Playgroud)

我收到以下错误:

AttributeError: 'str' object has no attribute 'predict'
Run Code Online (Sandbox Code Playgroud)

python automl

4
推荐指数
1
解决办法
7823
查看次数

标签 统计

automl ×1

python ×1