BigQueryML中用于TensorFlow模型的ML.PREDICT的多列输入

Posted 2021-04-03

我们已经在Google Cloud AutoML（我们非常喜欢的工具）中训练了一个模型并将其成功导出到GCS，然后使用以下命令在BigQuery中创建了模型：

create or replace model my_dataset.my_bq_ml_model
options(model_type='tensorflow',
 model_path='my gcs path to exported tensorflow model'))

但是，当我们使用BigQueryML尝试使用模型进行某些预测时，我们不确定如何将模型使用的多个功能格式化为导出的Tensorflow模型在BigQuery中接受的单个“输入”字符串。

select  *
from    ml.predict(model my_project.my_dataset.my_bq_ml_model,
(
 select 'How do we format this?'  as inputs
 from my_rows_to_predict
))

有人做过吗？

这类似于此问题，该问题仍然存在：Multi-column input to ML.PREDICT for a TensorFlow model in BigQuery ML

谢谢大家。

答案

将模型加载到BigQuery ML中后，在BigQuery UI中单击模型，然后切换到“模式”选项卡。这应该告诉您模型需要哪些列。

或者，在模型上运行程序save_model_cli（这是tensorflow附带的python程序，以查看受支持的签名是什么]

saved_model_cli show --dir $export_path --all