Skip to content

Commit

Permalink
Update README.md
Browse files Browse the repository at this point in the history
  • Loading branch information
tkipf authored Oct 28, 2018
1 parent a4823fd commit 4915806
Showing 1 changed file with 3 additions and 1 deletion.
4 changes: 3 additions & 1 deletion README.md
Original file line number Diff line number Diff line change
Expand Up @@ -4,7 +4,7 @@ This repository contains the official PyTorch implementation of:

**Neural relational inference for interacting systems.**
Thomas Kipf*, Ethan Fetaya*, Kuan-Chieh Wang, Max Welling, Richard Zemel.
https://arxiv.org/abs/1802.04687 (*: equal contribution)˚
https://arxiv.org/abs/1802.04687 (*: equal contribution)

![Neural Relational Inference (NRI)](nri.png)

Expand All @@ -24,6 +24,8 @@ python generate_dataset.py
```
This generates the springs dataset, use `--simulation charged` for charged particles.

Note: Make sure to use the same preprocessing and evaluation scripts (check the loss function as well) as in our code release to get comparable results.

### Run experiments

From the project's root folder, simply run
Expand Down

0 comments on commit 4915806

Please sign in to comment.