Code for the FiniteNet ICML Paper. To run this code, simply put the HelperFunctions.py file into your directory, along with any data you wish to use or trained models you wish to evaluate. Then run one of the attached scripts that start with 'test' or 'train' to test or train a FiniteNet model. To run this code, you must have numpy, pytorch, and matplotlib.
Dataset used to train for inviscid Burgers' equation: https://figshare.com/articles/invBurg_train_npy/11796201
Dataset used to train for KS equation: https://figshare.com/articles/ks_train_csv/11796198
Dataset used to test for KS equation: https://figshare.com/articles/ks_test_npy/11796186
Linear advection equation was trained and tested with data generated on the fly, as was data for testing inviscid Burgers' equation.