使用VertexAI ModelBatchPredictOp批量预测时遇ValueError问题求助

错误根源 #

报错的核心原因是ModelBatchPredictOp的model参数不接受VertexModel对象类型，该参数要求传入字符串类型的模型资源名称（格式如projects/{project_id}/locations/{location}/models/{model_id}），或是模型的GCS存储路径。

解决方法 #

从你已有的VertexModel对象中提取其资源名称字符串，替换原model参数的值即可：

修改后的代码示例 #

# 从VertexModel对象中获取模型资源名称
model_resource_name = model.resource_name

batch_predict_op = ModelBatchPredictOp(
    project=project_id,
    job_display_name="batch_prediction_job",
    model=model_resource_name,  # 替换为资源名称字符串
    gcs_source_uris=source_list,
    gcs_destination_output_uri_prefix=f"{bucket_name}",
    instances_format="csv",
    predictions_format="csv",
    #model_parameters={},
)

额外说明 #

• 无需修改gcs_source_uris的格式，保持gs://开头的URI列表即可，之前的调整方向有误
• 如果模型不在GCP默认区域，需在ModelBatchPredictOp中添加location参数，传入模型所在区域的字符串（如us-central1）

内容的提问来源于stack exchange，提问作者Paul