Plot losses

themightyoarfish · Mar 6, 2018 · 7fa1ab4 · 7fa1ab4
1 parent 7dbd535
commit 7fa1ab4
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/main.py b/main.py
@@ -135,8 +135,8 @@ def main():
             data_manager.shuffleBatches()
 
         f, ax = plt.subplots(1)
-        ax.plot(losses)
-        ax.set_title('Losses with 7/3 train-test split)')
+        ax.plot(np.arange(args.epochs), losses)
+        ax.set_title('Losses with 7/3 train-test split')
         ax.set_xlabel('epoch')
         ax.set_ylabel('Test loss')
         f.savefig('losses.pdf')