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)