Fixes

2021-05-15 01:15:56 +02:00 · 2021-05-15 01:15:56 +02:00 · b32605abe9
commit b32605abe9
parent 48fc3361df
2 changed files with 3 additions and 2 deletions
--- 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")
--- a/rmse.txt
+++ b/rmse.txt
@ -0,0 +1,2 @@
+0.10223432630300522
+0.12947769463062286