将 sklearn 模型提供给 AI Platform

问题描述

我正在尝试为 AI 平台提供一个简单的随机森林模型！

我已使用以下代码将其作为pickle文件保存在谷歌云存储中

with open('model.pkl','wb') as model_file:
    pickle.dump(model,model_file)

storage_client = storage.Client()
bucket = storage_client.bucket('bucket_name')
blob = bucket.blob("models/" + job_name + '/model.pkl')
blob.upload_from_filename('model.pkl')

模型在那个位置保存得很好。

为什么当我尝试在 AI 平台上提供服务时，会出现以下错误？有任何想法吗？错误不是很具体，它只声称 Error: 模型服务器从未准备好。请验证您的模型文件或容器配置是否有效。根据 GCP 文档，pickle 是一种有效的文件类型，所以我不明白我出了什么问题......提前致谢！！！ :D

解决方法

您的模型是使用自定义容器创建的吗？ 你是如何创建模型的？ 您是否指定了 --health-route=YOUR_HEALTH_ENDPOINT 和 --ports=PORT_WHERE_CONTAINER_LISTENS？