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

Name		Name	Last commit message	Last commit date
Latest commit History 1 Commit
Readme.md		Readme.md
modules.py		modules.py
train_fnri.py		train_fnri.py
train_nri.py		train_nri.py
train_rg.py		train_rg.py
train_rgi.py		train_rgi.py
utils.py		utils.py

Provide feedback