android - 如何将 Keras LSTM 模型用于 android？

Question

我使用 keras 创建了 LSTM 模型，并且我已经训练了存储为models.h5文件的模型。

我想知道如何在 Android Studio 中预测模型。

我怎么解决这个问题？

标签： androidpythonkeraslstm

Answer 1

您可以使用deeplearning4j做到这一点。您可以按照此处的说明在 Android Studio 中设置 deeplearning4j 。

可以使用此处描述的 Keras 模型导入功能将经过训练的 Keras 模型直接导入 deeplearning4j 。然后，您可以使用 Java 中的训练模型执行预测。