转换预训练的keras模型供H2O深水使用?

时间:2017-11-19 11:40:56

标签: tensorflow keras h2o

我有一个采用hdf5格式的Keras模型,我想用它来使用从H2O深水中进行预测。

(我正在使用Keras和Deep Water的tensorflow后端,并运行Deeper的Docker cpu版本,但如果必要的话我可以切换到gpu版本.Deep Water正在使用Tensorflow 1.1。)

我需要指定network_definition_filenetwork_parameters_file。在Deep Water booklet pdf的第6.5.2节中,前者具有.meta扩展名,而后者没有扩展名。

我发现https://stackoverflow.com/a/44349081/841830导致我https://github.com/amir-abdi/keras_to_tensorflow。它们似乎采用“.h5”文件并导出“.pb”文件。我可以同样使用“.hdf5”文件吗? “.pb”文件是否与深水文档中显示的无扩展名文件相同?

对于元文件,我认为我需要调用tf.train.export_meta_graph()函数,给它一个tf.train.Saver()对象。看起来cell 7 of old version是我能坚持的地方吗?

我不确定的最后一件事是如何将其用于预测。它与普通的H2O深度学习相同,你使用epochs=0技巧吗?如果使用Python API,我仍然需要调用train()或者是否足够创建H2ODeepWaterEstimator()对象?

0 个答案:

没有答案