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))