Uber Ludwig Toolbox:导出到Google Platform ML引擎的模型



在Uber Ludwig Toolbox中,从头开始产生以下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. triending_progress.p

我想将它们(或仅需要的)导出到一个文件 saved_model.pb 以便能够将其加载到我的ML Engine Google平台上。

我尝试了freeze_graph.py脚本,但是它需要作为我没有的输入a .pb文件。(我只有上述10个文件)。如何继续获取此 saved_model.pb 文件?我看到了几个线程,但是它们都不适合我。

ludwig增加了对SavedModel的支持

看这个问题

最新更新