Fix filterbank rCCA example and pipeline

thijor · Mar 22, 2024 · e6597dc · e6597dc
1 parent a91ebdc
commit e6597dc
Show file tree

Hide file tree

Showing 2 changed files with 4 additions and 2 deletions.
diff --git a/examples/fbrcca.py b/examples/fbrcca.py
@@ -94,7 +94,8 @@
         # Setup classifier
         rcca = pyntbci.classifiers.rCCA(stimulus=V, fs=fs, event=event, encoding_length=encoding_length,
                                         onset_event=onset_event)
-        fbrcca = pyntbci.classifiers.Ensemble(estimator=rcca, gating="mean")
+        gate = pyntbci.gating.AggregateGate("mean")
+        fbrcca = pyntbci.classifiers.Ensemble(estimator=rcca, gating=gate)
 
         # Train classifier
         fbrcca.fit(X_trn, y_trn)

diff --git a/pipelines/thielen2021_fbrcca.py b/pipelines/thielen2021_fbrcca.py
@@ -97,7 +97,8 @@
         # Setup classifier
         rcca = pyntbci.classifiers.rCCA(stimulus=V, fs=fs, event=event, encoding_length=encoding_length,
                                         onset_event=onset_event)
-        fbrcca = pyntbci.classifiers.Ensemble(estimator=rcca, gating="mean")
+        gate = pyntbci.gating.AggregateGate("mean")
+        fbrcca = pyntbci.classifiers.Ensemble(estimator=rcca, gating=gate)
 
         # Train classifier
         fbrcca.fit(X_trn, y_trn)