update readme

ethanfetaya · Feb 14, 2018 · f31ad4b · f31ad4b
1 parent f6eb28f
commit f31ad4b
Showing 1 changed file with 8 additions and 0 deletions.
diff --git a/README.md b/README.md
@@ -1,4 +1,12 @@
 # NRI
 Neural relational inference for interacting systems - pytorch
 
+### Data generation
 
+cd data
+
+python generate_dataset.py
+
+cd ..
+
+That generates the springs dataset, use '--simulation charged' for charged particles.