Update 'pred.py'

This commit is contained in:
Wojciech Mikołajski 2023-06-17 20:37:56 +02:00
parent 4b08f485f7
commit 76cdf71383

View File

@ -25,7 +25,7 @@ prediction_classes = [1 if prob > 0.5 else 0 for prob in np.ravel(predictions)]
print(prediction_classes[:30]) print(prediction_classes[:30])
with open("predictionsResults.txt", mode='w', newline='') as f: with open("predictionsResults.txt", mode='w', newline='') as f:
f.write(str(f'Results:)) f.write(str(f'Results:'))
for pred in predictions: for pred in predictions:
f.write(str(f'{pred[0]}')) f.write(str(f'{pred[0]}'))
f.write("\n") f.write("\n")