24 lines
554 B
Python
24 lines
554 B
Python
import pandas
|
|
import os
|
|
|
|
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')
|
|
|
|
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') |