使用CSV输入的Vertex AI批处理预测



我们一直在尝试使用存储在GCS中的CSV作为输入和存储在GCS中的模型pickle文件来运行批处理预测。然而,不幸的是,我们似乎无法让预测与CSV输入一起工作。

我们创建了一个测试CSV,它是一个16x16的单位矩阵,带有一个标题行。

,

input_1,input_2,input_3
1.0,0.0,0.0
0.0,1.0,0.0
0.0,0.0,1.0 

文件的格式是这样的,有更多的颜色和行。

我们的模型预计有16个特征。

使用这个CSV和GCS中的Model Pickle,我们从控制台UI启动了一个批处理预测请求。

作业运行了一会儿,在管道UI中显示成功,但是没有返回结果,并且在GCS中产生的错误文件为空。

我检查了顶点预测文档,似乎我的CSV格式与那里演示的相同。

使用这个模型,但是使用格式化的JSON输入-我们确实得到了一个结果,所以我不认为它是模型。

这是已知的问题吗?

是否有人成功地在顶点AI上运行批量预测与CSV作为输入?

似乎需要给列加引号,除非这些列名是数字。
所以你需要引用你的标题:

"input_1","input_2","input_3"

在顶点AI批处理请求输入文档中有一个错误,如所示的示例没有引号,文档应该很快更新(我已经报告了)。

最近确实发布了一个修复,但它必须回滚,目前还不清楚这个报价要求是否最终会改变。

这是一个产品bug。谷歌上周发布了一个修复程序。请重新测试CSV,它应该工作。

最新更新