как модели Keras LSTM на андроид

Question

Я использовал karas для создания модели LSTM, и я уже обучил модели, которые хранятся в виде models.h5 файлов.

Я хочу знать, как прогнозировать модель в студии Android.

как я могу решить эту проблему?

Answer 1

Вы можете сделать это, используя deeplearning4j . Вы можете настроить deeplearning4j в Android Studio, следуя инструкциям здесь .

Обученные модели Keras можно импортировать непосредственно в deeplearning4j с помощью функции импорта модели Keras, описанной здесь . Затем вы можете выполнять прогнозы с помощью обученной модели на Java.