/keras-minimal-rnn

Keras implementation of MinimalRNN: Toward More Interpretable and Trainable Recurrent Neural Networks

Primary LanguagePythonMIT LicenseMIT

Keras MinimalRNN

Keras implementation of MinimalRNN: Toward More Interpretable and Trainable Recurrent Neural Networks.

Network graph of MinimalRNNs (from the paper):

Usage

Import minimal_rnn.py and use either the MinimalRNNCell or MinimalRNN layer

from minimal_rnn import MinimalRNN 

# this imports the layer rather than the cell
ip = Input(...)  # Rank 3 input shape
x = MinimalRNN(units=128)(ip)
...