我有一个采用hdf5格式的Keras模型,我想用它来使用从H2O深水中进行预测。
(我正在使用Keras和Deep Water的tensorflow后端,并运行Deeper的Docker cpu版本,但如果必要的话我可以切换到gpu版本.Deep Water正在使用Tensorflow 1.1。)
我需要指定network_definition_file
和network_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()
对象?