From 6b8c43b7c9033c5c9729d44aa382a10025770b15 Mon Sep 17 00:00:00 2001
From: Wojciech Jarmosz <wojciech@jarmosz.com>
Date: Fri, 14 May 2021 04:16:17 +0200
Subject: [PATCH] Fix metrics

---
 Jenkins_eval  | 2 +-
 evaluation.py | 7 ++-----
 2 files changed, 3 insertions(+), 6 deletions(-)

diff --git a/Jenkins_eval b/Jenkins_eval
index 4f29a84..7b4dd03 100644
--- a/Jenkins_eval
+++ b/Jenkins_eval
@@ -27,7 +27,7 @@ pipeline {
 		}
         stage('Save metrics to bulk file & create chart') {
             steps{
-                sh "if [ -f "single_metrics.txt" ]; "${env.BUILD_ID} ${cat single_metrics.txt}" >> bulk_metrics.txt fi"
+                sh "if [ -f "rmse.txt" ]; "${env.BUILD_ID} ${cat rmse.txt}" >> bulk_metrics.txt fi"
                 archiveArtifacts 'bulk_metrics.txt'
                 sh "python3 create_chart.py"
                 archiveArtifacts 'chart.png'
diff --git a/evaluation.py b/evaluation.py
index 8e13373..1cc7dc2 100644
--- a/evaluation.py
+++ b/evaluation.py
@@ -20,9 +20,6 @@ model = tf.keras.models.load_model('linear_regression.h5')
 scores = model.evaluate(x=test_X, 
                     y=test_Y)    
 
-with open('single_metrics.txt', 'w') as file:
+with open('rmse.txt', 'w') as file:
     for idx, score in enumerate(scores):
-        if idx == 0:
-            file.write("Recall: " + str(score) + "\n")
-        if idx == 1:
-            file.write("RMSE: " + str(score) + "\n")
+        file.write(str(score))