This repository contains the code for this ArXiv paper: https://arxiv.org/abs/1801.04342
File equationGenerator.ipynb is an iPython notebook that describes the dataset generation method proposed in the paper and lstmTreeNotebook.ipynb contains the code for training over the generated equations.