Update 'predict.py'

This commit is contained in:
Patryk Gałka 2023-05-10 20:41:43 +02:00
parent 683e7592b7
commit 732babcc90

View File

@ -14,5 +14,8 @@ y_to_test = pandas.read_csv('./Y_test.csv')
accu = reloaded.evaluate(x_to_test, y_to_test) accu = reloaded.evaluate(x_to_test, y_to_test)
pre = reloaded.predict(x_to_test) pre = reloaded.predict(x_to_test)
pre.tofile('prediction.csv', sep=',', format='%s') pre_array = [float(i)/sum(pre) for i in pre]
pre_array.tofile('prediction.csv', sep=',', format='%s')