Update preprocess_test.py

preprocess_test.py

@@ -136,7 +136,8 @@ class Preprocess_Test:
         BATCH_SIZE = 256
         correct = 0
         total = 0
-
+        all_predictions = []
+        
         with torch.no_grad():
             for i in range(0, len(X_test_t), BATCH_SIZE):
                 batch_x = X_test_t[i:i + BATCH_SIZE].to(self.device)
@@ -146,12 +147,15 @@ class Preprocess_Test:
                 predicted = torch.argmax(outputs, dim=1)
                 total += batch_y.size(0)
                 correct += (predicted == batch_y).sum().item()
-
+                all_predictions.extend(predicted.cpu().numpy().tolist())
+               
                 if i == 0:
                     print(f"Test batch - Predicted: {predicted.cpu().numpy()[:10]}")
                     print(f"Test batch - Actual: {batch_y.cpu().numpy()[:10]}")
 
 
-                    
+        return {
+        "predictions": all_predictions}
+