From d23492d028038a2f6a51ae8bc1cf03486aff9b3f Mon Sep 17 00:00:00 2001
From: Sheaza <nataliawolowiec3@gmail.com>
Date: Tue, 14 May 2024 22:17:35 +0200
Subject: [PATCH] add accuracy

---
 evaluate/Jenkinsfile | 2 +-
 predict.py           | 5 +++++
 2 files changed, 6 insertions(+), 1 deletion(-)

diff --git a/evaluate/Jenkinsfile b/evaluate/Jenkinsfile
index b780003..29bc5d2 100644
--- a/evaluate/Jenkinsfile
+++ b/evaluate/Jenkinsfile
@@ -30,7 +30,7 @@ pipeline {
             steps {
                 sh "chmod +x ./predict.py"
                 sh "python ./predict.py"
-                archiveArtifacts artifacts: 'predictions.txt', onlyIfSuccessful: true
+                archiveArtifacts artifacts: 'predictions.txt,accuracy.txt', onlyIfSuccessful: true
             }
         }
     }
diff --git a/predict.py b/predict.py
index 7ac8c47..990305e 100644
--- a/predict.py
+++ b/predict.py
@@ -1,6 +1,7 @@
 import pandas as pd
 from tensorflow import keras
 import numpy as np
+from sklearn.metrics import accuracy_score
 np.set_printoptions(threshold=np.inf)
 
 data = pd.read_csv("df_test.csv")
@@ -13,3 +14,7 @@ predictions = model.predict(X_test)
 
 with open("predictions.txt", "w") as f:
     f.write(str(predictions))
+
+accuracy = accuracy_score(y_test, predictions)
+with open("accuracy.txt", 'a') as file:
+    file.write(str(accuracy))