Code changes

brain2vec_PCA.py

@@ -237,10 +237,10 @@ def main():
     print(f"Saved reconstructions to {recons_path}")
 
     # Optionally save the actual PCA model with joblib
-    # from joblib import dump
-    # ipca_model_path = os.path.join(args.output_dir, "pca_model.joblib")
-    # dump(model.ipca, ipca_model_path)
-    # print(f"Saved PCA model to {ipca_model_path}")
+    from joblib import dump
+    ipca_model_path = os.path.join(args.output_dir, "pca_model.joblib")
+    dump(model.ipca, ipca_model_path)
+    print(f"Saved PCA model to {ipca_model_path}")
 
 
 if __name__ == "__main__":