Test Client: Improve prediction success visibility

softawaregmbh · Nov 16, 2022 · 33b440e · 33b440e
1 parent 0da486d
commit 33b440e
Showing 1 changed file with 2 additions and 1 deletion.
diff --git a/src/4_test_client.py b/src/4_test_client.py
@@ -79,7 +79,8 @@ def pretty_print_results(pred_samples, y_samples):
           '├───────┼────────────┤')
     if pred_samples is not None:
         for label, prediction in zip(y_samples, pred_samples):
-            print(f'│ {label:<5} │ {prediction}          │')
+            success_label = '✅' if label == prediction else '❌'
+            print(f'│ {label:<5} │ {prediction}          │ {success_label}')
     print('└───────┴────────────┘')