ium_z434743/prediction.py

21 lines
453 B
Python

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')