README.md

Neural Arithmetic Logic Units

MXNet implementation of Neural Arithmetic Logic Units by Andrew Trask, Felix Hill, Scott Reed, Jack Rae, Chris Dyer and Phil Blunsom using MNIST dataset

Training

from mxnet import *

# Training and saving the model
python nalu.py action train

Prediction

from mxnet import *

# Performing prediction using the saved mxnet model
python nalu.py action predict

Note : Default action is train: python nalu.py

TODO: Reinforcement Learning Experiments