diff --git a/prediction.py b/prediction.py new file mode 100644 index 0000000..d7cd35e --- /dev/null +++ b/prediction.py @@ -0,0 +1,20 @@ +import pandas +import os +import tensorflow + +build_number = int(os.environ['BUILD_NUMBER']) + +reloaded = tensorflow.keras.models.load_model('test') + +x_to_test = pandas.read_csv('./X_test.csv') +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'{build_number},{accu}\n') + +pre = reloaded.predict(x_to_test) + +pre.tofile('prediction.csv', sep=',', format='%s')