Learning RNN Hierarchies
about.md
has the basic details of what the code is trying to do and why.
- Layers are defined in
layers/
along with tests - All layers derive from abstract base class defined in
base.py
- All data and data prepping scripts are in
data/
folder network.py
has tools for taking in a list as a model and doing layer by layer forward/backward pass, getting gradients, setting/getting parameterstrain_ptb.py
trains a model on Penn Tree Bank text file, which has to be placed in thedata/
foldertrain_mnist.py
trains a model on Sequential MNIST.mnist.pkl.gz
has to be placed indata/
folder- As the network trains, logs are generated. Final logs and models are stored as pickle objects in
results/experiment_name
, whereexperiment_name
is a string defined intrain_
scripts
##Requires:
- Numpy
- Scipy (for one special function to calculate entropy)
- matplotlib
- climin