# RoeNets: Predicting Discontinuity of Hyperbolic Systems from Continuous Data

This repository contains the code supplement for the ICLR 2021 submission "RoeNets: Predicting Discontinuity of Hyperbolic Systems from Continuous Data." For each experimental task outlined in the paper, we include datasets, training code, and trained models. 

gen_data.py will generate the dataset and save it as 'data.h5'
train_net.py will train the RoeNet and save the model as 'model.pt' 

There are no major dependencies besides PyTorch (tested with version 1.5.0). All code tested with Python 3.7.6.

