如何将Keras LSTM模型转换为Android

Question

我使用karas创建LSTM模型，并且我已经训练了存储为implementation文件的模型。

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

我该如何解决这个问题？

Answer 1

您可以使用deeplearning4j进行此操作。您可以按照here的说明在Android Studio中设置deeplearning4j。

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