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 Platform（Cloud ML Engine）支持的唯一模型格式

看看这个问题