Uber Ludwig Toolbox：将生成的模型导出到Google平台ML Engine

Question

在Uber Ludwig工具箱中，从头开始的培训产生了以下10个文件：

1. 检查点
2. model_hyperparameters.json
3. model_weights.data-00000-of-00001
4. model_weights.index
5. model_weights.meta
6. model_weights_progress.data-00000-of-00001
7. model_weights_progress.index
8. model_weights_progress.meta
9. train_set_metadata.json
10. training_progress.p

我想将它们（或仅将所需的文件）导出到一个文件 saved_model.pb 中，以便将其加载到我的ML Engine Google平台上。

我尝试了freeze_graph.py脚本，但是它需要输入一个我没有的.pb文件作为输入。 （我只有上述10个文件）。如何继续获取这个 saved_model.pb 文件？我看到了几个线程，但是没有一个适合我。

Answer 1

Ludwig添加了对SavedModel的支持，这是AI平台（Cloud ML Engine）唯一支持的模型格式。

看看这个issue