From 09cbfad0efe0cbd891b7a31fbae2d7cf5810cc59 Mon Sep 17 00:00:00 2001
From: jakubknczny <jakubknczny@gmail.com>
Date: Sun, 16 May 2021 16:06:07 +0200
Subject: [PATCH] just loss

---
 lab5/train/train.py | 2 +-
 1 file changed, 1 insertion(+), 1 deletion(-)

diff --git a/lab5/train/train.py b/lab5/train/train.py
index b8ca4c9..1f3faeb 100644
--- a/lab5/train/train.py
+++ b/lab5/train/train.py
@@ -46,7 +46,7 @@ def prepare_train_model(learning_rate, _run):
 
     model.save('grid-stability-dense.h5')
 
-    _run['history'] = str(history.history[:, -1])
+    _run['history'] = str(history.history['loss'][1])
 
 
 @ex.main