ium_z434686/predict.py

22 lines
499 B
Python

import pandas
import os
from keras.applications.densenet import layers
from sklearn.model_selection import train_test_split
import tensorflow
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.txt', 'w') as file:
file.write(f'Accuracy: {accu}')
pre = reloaded.predict(x_to_test)
pre.tofile('prediction.csv', sep=',', format='%s')