ium_z434686/predict.py

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}")