Equation-AI

View the colab notebook at:

https://colab.research.google.com/drive/1_ZSqKAsMqcfPFW2L8jlRJpkxpVcnKEIS?usp=sharing

View samples in the samples folder

To train the model (this will take a long time)

pip install -r requirements.txt

python get_data.py

python train_model.py

To sample a pre-existing model (this will take a long time)

pip install -r requirements.txt

python sample_model.py

Contribution

get_data.py: Bo Zhang, Cristobal Benavides

PixelCNN.py: Bo Zhang, Seamus Sutula

train_model.py/sample_model.py: Cristobal Benavides, Seamus Sutula