Update train.py

Author: AdarshMJ

Spectral/train.py

@@ -82,7 +82,7 @@ def test():
     print(f"Average Test Accuracy: {np.mean(test_acc_allsplits):.2f}% ± {2 * np.std(test_acc_allsplits) / np.sqrt(len(test_acc_allsplits)):.2f}%")
     print(f"Average Validation Accuracy: {np.mean(val_acc_allsplits):.2f}% ± {2 * np.std(val_acc_allsplits) / np.sqrt(len(val_acc_allsplits)):.2f}%")
 
-    return final_test_acc, 2 * np.std(test_acc_allsplits) / np.sqrt(len(test_acc_allsplits)),final_val_acc,2 * np.std(val_acc_allsplits) / np.sqrt(len(val_acc_allsplits))
+    return np.mean(test_acc_allsplits) , 2 * np.std(test_acc_allsplits) / np.sqrt(len(test_acc_allsplits)),final_val_acc,2 * np.std(val_acc_allsplits) / np.sqrt(len(val_acc_allsplits))