improve readme

jeiros · May 24, 2018 · 8731cb0 · 8731cb0
1 parent cc0f6ad
commit 8731cb0
Show file tree

Hide file tree

Showing 4 changed files with 14 additions and 20 deletions.
diff --git a/README.md b/README.md
@@ -27,5 +27,8 @@ from mdgan import MDGAN
 trjs = AlanineDipeptide().get().trajectories
 data, sc = make_trajectory_trainable(trjs)  # sc is the MinMaxScaler we'll need it later
 gan = MDGAN(n_atoms=22)
-losses = gan.train(data, num_epochs=10)  # That's it
-```
+losses = gan.train(data, num_epochs=50)  # That's it
+```
+
+![losses](./losses.png)
+![features](./features.png)