From a23a72f568da6fd0cc4224790507812d997cb05e Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Thu, 11 May 2023 00:31:05 +0200 Subject: [PATCH] Update 'predict.py' --- predict.py | 4 +++- 1 file changed, 3 insertions(+), 1 deletion(-) diff --git a/predict.py b/predict.py index 4ca07a0..b8237dd 100644 --- a/predict.py +++ b/predict.py @@ -6,6 +6,7 @@ from keras.applications.densenet import layers from sklearn.model_selection import train_test_split import tensorflow +build_number = int(os.environ['BUILD_NUMBER']) reloaded = tensorflow.keras.models.load_model('test') @@ -14,8 +15,9 @@ y_to_test = pandas.read_csv('./Y_test.csv') accu = reloaded.evaluate(x_to_test, y_to_test) + with open('metrics.csv', 'a') as file: - file.write(f'{accu},') + file.write(f'{build_number},{accu}\n') pre = reloaded.predict(x_to_test)