diff --git a/pred.py b/pred.py index 7a8692d..babf04e 100644 --- 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")