From 7968db013bb25b4d74f621ed700ac9c594d29aec Mon Sep 17 00:00:00 2001
From: =?UTF-8?q?Patryk=20Ga=C5=82ka?= <patgal2@st.amu.edu.pl>
Date: Wed, 10 May 2023 20:44:28 +0200
Subject: [PATCH] Update 'predict.py'

---
 predict.py | 4 +++-
 1 file changed, 3 insertions(+), 1 deletion(-)

diff --git a/predict.py b/predict.py
index 4b36927..bc62418 100644
--- a/predict.py
+++ b/predict.py
@@ -18,4 +18,6 @@ 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]
 
-pre_array.tofile('prediction.csv', sep=',', format='%s')
\ No newline at end of file
+with open("prediction.txt", "w") as txt_file:
+    for line in pre_array:
+        txt_file.write(f"{line}")
\ No newline at end of file