Add resetStates function for stateful rnns

transcranial / keras-js

Run Keras models in the browser, with GPU support using WebGL

https://transcranial.github.io/keras-js

MIT License

4.96k stars 501 forks source link

Add resetStates function for stateful rnns #98

Open mattdangerw opened 6 years ago

mattdangerw commented 6 years ago

No idea if this is of interest, but working on some web demos for rnn text generation with stateful rnns and sorely missed the reset_states function from keras. Added one in.

Probably plenty of ways to implement this too so happy to tweak.