Big update

jameschapman19 · Oct 26, 2023 · 35b2636 · 35b2636
1 parent ed6419c
commit 35b2636
Showing 1 changed file with 2 additions and 2 deletions.
diff --git a/docs/source/examples/plot_dcca_multi.py b/docs/source/examples/plot_dcca_multi.py
@@ -40,7 +40,7 @@
 dcca_mcca = DCCA(
     latent_dimensions=LATENT_DIMS,
     encoders=[encoder_1, encoder_2],
-    objective=objectives.MCCA,
+    objective=objectives.MCCALoss,
 )
 trainer_mcca = pl.Trainer(max_epochs=EPOCHS, enable_checkpointing=False, enable_model_summary=False,enable_progress_bar=False)
 trainer_mcca.fit(dcca_mcca, train_loader, val_loader)
@@ -54,7 +54,7 @@
 dcca_gcca = DCCA(
     latent_dimensions=LATENT_DIMS,
     encoders=[encoder_1, encoder_2],
-    objective=objectives.GCCA,
+    objective=objectives.GCCALoss,
 )
 trainer_gcca = pl.Trainer(max_epochs=EPOCHS, enable_checkpointing=False, enable_model_summary=False,enable_progress_bar=False)
 trainer_gcca.fit(dcca_gcca, train_loader, val_loader)