Update 'pred.py'

2023-06-17 20:14:19 +02:00 · 2023-06-17 20:14:19 +02:00 · 4c8e952dca
commit 4c8e952dca
parent 3059d0378b
1 changed files with 1 additions and 0 deletions
--- a/pred.py
+++ b/pred.py
@ -25,6 +25,7 @@ prediction_classes = [1 if prob > 0.5 else 0 for prob in np.ravel(predictions)]
 print(prediction_classes[:30])

 with open("predictionsResults.txt", mode='w', newline='') as f:
+    f.write(str(f'Results:))
    for pred in predictions:
        f.write(str(f'{pred[0]}'))
        f.write("\n")