add accuracy

This commit is contained in:
Sheaza 2024-05-14 22:17:35 +02:00
parent edfca60b98
commit d23492d028
2 changed files with 6 additions and 1 deletions

View File

@ -30,7 +30,7 @@ pipeline {
steps { steps {
sh "chmod +x ./predict.py" sh "chmod +x ./predict.py"
sh "python ./predict.py" sh "python ./predict.py"
archiveArtifacts artifacts: 'predictions.txt', onlyIfSuccessful: true archiveArtifacts artifacts: 'predictions.txt,accuracy.txt', onlyIfSuccessful: true
} }
} }
} }

View File

@ -1,6 +1,7 @@
import pandas as pd import pandas as pd
from tensorflow import keras from tensorflow import keras
import numpy as np import numpy as np
from sklearn.metrics import accuracy_score
np.set_printoptions(threshold=np.inf) np.set_printoptions(threshold=np.inf)
data = pd.read_csv("df_test.csv") data = pd.read_csv("df_test.csv")
@ -13,3 +14,7 @@ predictions = model.predict(X_test)
with open("predictions.txt", "w") as f: with open("predictions.txt", "w") as f:
f.write(str(predictions)) f.write(str(predictions))
accuracy = accuracy_score(y_test, predictions)
with open("accuracy.txt", 'a') as file:
file.write(str(accuracy))