fix: overwrite: test_accuracy.png

Author: raselmandol

utils.py

@@ -67,19 +67,20 @@ def save_test_samples_images(model, dataloader, device, out_dir=".", num_samples
 
     return results
 
-def save_test_accuracy_image(test_accuracy: float, out_dir: str = ".") -> str:
-    """Create/overwrite an image summarizing test accuracy as text."""
+def save_test_accuracy_image(test_accuracies, out_dir: str = ".") -> str:
+    """
+    Create/overwrite an image summarizing test accuracy as a graph, similar to train_accuracy.png.
+    Accepts a list of test accuracies per batch or sample.
+    """
     os.makedirs(out_dir, exist_ok=True)
-    plt.figure(figsize=(4, 3))
-    plt.axis('off')
-    plt.text(0.5, 0.6, 'Test Accuracy', ha='center', va='center', fontsize=16, weight='bold')
-    plt.text(0.5, 0.35, f"{test_accuracy * 100:.2f}%", ha='center', va='center', fontsize=22)
-    # Optional simple bar
-    acc = max(0.0, min(1.0, float(test_accuracy)))
-    plt.hlines(0.1, 0.1, 0.9, colors='#e0e0e0', linewidth=12)
-    plt.hlines(0.1, 0.1, 0.1 + 0.8 * acc, colors='#4caf50', linewidth=12)
+    plt.figure()
+    plt.plot(range(1, len(test_accuracies)+1), test_accuracies, marker='o', color='#4caf50')
+    plt.title('Test Accuracy')
+    plt.xlabel('Batch')
+    plt.ylabel('Accuracy')
+    plt.grid(True)
     out_path = os.path.join(out_dir, 'test_accuracy.png')
-    plt.savefig(out_path, bbox_inches='tight')
+    plt.savefig(out_path)
     plt.close()
     return out_path
 def predict_and_log_samples(model, dataloader, device, num_samples=5):