ium_z434686/predict.py

23 lines
525 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)
pre = reloaded.predict(x_to_test)
pre_array = [float(i)/sum(pre) for i in pre]
with open("prediction.txt", "w") as txt_file:
for line in pre_array:
txt_file.write(f"{line}")