Состояние сброса Tensorflow Java RNN-LSTM - PullRequest
0 голосов
/ 13 декабря 2018

Можно ли сбросить скрытое состояние RNN-LSTM с помощью API-интерфейса Tensorflow Java?В настоящее время я прогнозирую, используя загруженную модель;

model = SavedModelBundle.load(MODEL_PATH, "serve");
outputs = model.session().runner().feed("input_layer", in).fetch("time_distributed/transpose_1:0").run();

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

model.reset_states()

Мне нужно использовать модель stateful=True и затем сбросить ее состояние.Отправка полной последовательности и использование stateful=False не является решением для меня из-за проблем с производительностью.

...