在Google vertex ai上创建自定义模型

Question

我应该使用谷歌的托管ML平台Vertex AI为实习建立一个端到端的机器学习工作流程。尽管我完全遵循本教程，但在运行培训作业时，我会看到以下错误消息：

Training pipeline failed with error message: There are no files under "gs://dps-fuel-bucket/mpg/model" to copy.

根据本教程，我们不应该在存储桶中包含/model目录。模型应创建此目录并将最终结果保存在该目录中。

# Export model and save to GCS
model.save(BUCKET + '/mpg/model')

我添加了这个目录，但仍然会遇到这个错误。有没有人知道，先说声谢谢:)

Answer 1

如果您使用的是预先构建的容器，请确保您的模型工件具有与以下示例完全匹配的文件名：

TensorFlow SavedModel: saved_model.pb
scikit-learn: model.joblib or model.pkl
XGBoost: model.bst, model.joblib, or model.pkl

参考：Vertex-AI Model Import