Update 'predict.py'

This commit is contained in:
Patryk Gałka 2023-05-10 20:49:17 +02:00
parent 433b6b6c0b
commit bc26ba7528

View File

@ -16,8 +16,5 @@ 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}")
pre.tofile('prediction.csv', sep=',', format='%s')