From b32605abe9380769ff81448d99714d49bb90e61c Mon Sep 17 00:00:00 2001
From: Wojciech Jarmosz <wojciech@jarmosz.com>
Date: Sat, 15 May 2021 01:15:56 +0200
Subject: [PATCH] Fixes

---
 evaluation.py | 3 +--
 rmse.txt      | 2 ++
 2 files changed, 3 insertions(+), 2 deletions(-)
 create mode 100644 rmse.txt

diff --git a/evaluation.py b/evaluation.py
index bb2a898..0e34c72 100644
--- a/evaluation.py
+++ b/evaluation.py
@@ -21,5 +21,4 @@ scores = model.evaluate(x=test_X,
                     y=test_Y)    
 
 with open('rmse.txt', 'w') as file:
-    for idx, score in enumerate(scores):
-        file.write(str(score) + "\n")
+    file.write(str(scores[1]) + "\n")
diff --git a/rmse.txt b/rmse.txt
new file mode 100644
index 0000000..01c2c36
--- /dev/null
+++ b/rmse.txt
@@ -0,0 +1,2 @@
+0.10223432630300522
+0.12947769463062286