Update 'predict.py'

This commit is contained in:
Patryk Gałka 2023-05-10 20:44:28 +02:00
parent 732babcc90
commit 7968db013b

View File

@ -18,4 +18,6 @@ accu = reloaded.evaluate(x_to_test, y_to_test)
pre = reloaded.predict(x_to_test) pre = reloaded.predict(x_to_test)
pre_array = [float(i)/sum(pre) for i in pre] pre_array = [float(i)/sum(pre) for i in pre]
pre_array.tofile('prediction.csv', sep=',', format='%s') with open("prediction.txt", "w") as txt_file:
for line in pre_array:
txt_file.write(f"{line}")