From 653939737b25c2cf26d2da4f001c9da64a64ed13 Mon Sep 17 00:00:00 2001 From: Klaudia Date: Thu, 11 May 2023 18:32:15 +0200 Subject: [PATCH] fix --- Jenkinsfile_stats | 2 +- predictions.py | 15 +++++++++++---- 2 files changed, 12 insertions(+), 5 deletions(-) diff --git a/Jenkinsfile_stats b/Jenkinsfile_stats index 1f590bc..215dda2 100644 --- a/Jenkinsfile_stats +++ b/Jenkinsfile_stats @@ -19,7 +19,7 @@ pipeline { } stage('copy_artifacts') { steps { - copyArtifacts filter: 'adult_test.csv,adult_dev.csv,adult_train.csv', fingerprintArtifacts: true, projectName: 'z-s444439-create-dataset', selector: workspace() + copyArtifacts filter: 'X_test.csv,X_dev.csv,X_train.csv', fingerprintArtifacts: true, projectName: 'z-s444439-create-dataset', selector: workspace() } } stage('Docker') { diff --git a/predictions.py b/predictions.py index d3ddf95..0fae43b 100644 --- a/predictions.py +++ b/predictions.py @@ -3,10 +3,17 @@ import numpy as np import tensorflow as tf import pandas as pd -test_data = pd.read_csv("adult_test.csv") +data = tf.keras.models.load_model('model') -model = keras.models.load_model("model.h5") +x_to_test = pd.read_csv('./X_test.csv') +y_to_test = pd.read_csv('./Y_test.csv') -predictions = model.predict(test_data) +accu = data.evaluate(x_to_test, y_to_test) -np.savetxt("predictions.csv", predictions, delimiter=",") + +with open('metrics.csv', 'a') as file: + file.write(f'1,{accu}\n') + +pre = data.predict(x_to_test) + +pre.tofile('prediction.csv', sep=',', format='%s')