Conditional Neural Relational Inference for Interacting Systems Code for the cNRI model. Paper link. To run the code: python train_nri.py or python train_fnri.py or python train_pg.py or python train_ig.py