我已经使用 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)