From bc26ba75289f9bbbcb5bdbf41ee2b0a968d8b117 Mon Sep 17 00:00:00 2001 From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= Date: Wed, 10 May 2023 20:49:17 +0200 Subject: [PATCH] Update 'predict.py' --- predict.py | 5 +---- 1 file changed, 1 insertion(+), 4 deletions(-) diff --git a/predict.py b/predict.py index bc62418..de75e9b 100644 --- a/predict.py +++ b/predict.py @@ -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}") \ No newline at end of file +pre.tofile('prediction.csv', sep=',', format='%s') \ No newline at end of file