diff --git a/predict.py b/predict.py
index 7f01e14..4b36927 100644
--- a/predict.py
+++ b/predict.py
@@ -14,5 +14,8 @@ y_to_test = pandas.read_csv('./Y_test.csv')
 
 accu = reloaded.evaluate(x_to_test, y_to_test)
 
+
 pre = reloaded.predict(x_to_test)
-pre.tofile('prediction.csv', sep=',', format='%s')
\ No newline at end of file
+pre_array = [float(i)/sum(pre) for i in pre]
+
+pre_array.tofile('prediction.csv', sep=',', format='%s')
\ No newline at end of file