google-cloud-platform - 在 Google vertex ai 上创建自定义模型

Question

我应该使用 Google 的托管 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

If you're using a pre-built container, ensure that your model artifacts have filenames that exactly match the following examples:

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

Reference : Vertex-AI Model Import