fferroni / PhasedLSTM-Keras

Keras implementation of Phased LSTM [https://arxiv.org/abs/1610.09513]

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

PhasedLSTM-Keras

Keras implementation of Phased LSTM [https://arxiv.org/abs/1610.09513], from NIPS 2016.

To install run: `pip install phased_lstm_keras`

Works both with Theano and Tensorflow backend (tested on Theano 0.9 and Tensorflow 1.0).

Classification performance compared to standard Keras LSTM for MNIST dataset:

Accuracy [red: PLSTM, black: LSTM]

Loss [red: PLSTM, black: LSTM]

About

Keras implementation of Phased LSTM [https://arxiv.org/abs/1610.09513]

License:MIT License


Languages

Language:Python 100.0%