如何以不同格式保存keras模型

Question

我是这个领域的新手，你们中的一些人可能觉得这个问题很傻，但请大家耐心点。

我创建了一个运行良好的keras模型。我可以用.hdf5格式保存它，这是很好的model.save("model_name.hdf5)。

主要问题是，如果有任何其他格式，我可以保存我的模型，以便它可以用于c++/android/JavaScript。这有可能做到吗？如果你在想我为什么要求所有3种语言，那是因为我有3个项目，每个项目都使用各自的语言。

谢谢你提前提供帮助。

Answer 1

答案取决于您要在另一种语言中保存和使用什么。

如果您只需要保存模型的体系结构，可以将其保存为JSON，以后可以在您将要使用的任何其他平台和语言中使用它。

model_json = model.to_json()

如果您还需要权重和偏差，我不知道任何特定的工具，但是您可以简单地读取python中存储的数据，创建多维数组，然后将其保存在适合您需要的任何语言的文件中。例如，第二层的权重可以在model.layers[2].get_weights()中找到。

最后，如果您想以另一种语言运行模型，则需要实现进行处理的for-循环。您可能会为目标语言找到一些转换工具。例如，对于C，这个工具可以提供帮助。