radiata-ai
/

brain2vec_PCA

Model card Files Files and versions Community

jesseab commited on Feb 13

Commit

31eae02

·

1 Parent(s): 653468a

Code changes

Files changed (1) hide show

brain2vec_PCA.py +4 -4

brain2vec_PCA.py CHANGED Viewed

@@ -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}")
 if __name__ == "__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}")
 if __name__ == "__main__":