Skip to content

Commit 590feb3

Browse files
committed
CSP: only log once if cov is computed epoch wise
1 parent dda454d commit 590feb3

2 files changed

Lines changed: 10 additions & 0 deletions

File tree

mne/cov.py

Lines changed: 2 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -2108,6 +2108,7 @@ def regularize(
21082108
return cov
21092109

21102110

2111+
@verbose
21112112
def _regularized_covariance(
21122113
data,
21132114
reg=None,
@@ -2118,6 +2119,7 @@ def _regularized_covariance(
21182119
log_ch_type=None,
21192120
log_rank=None,
21202121
cov_kind="",
2122+
verbose=None,
21212123
):
21222124
"""Compute a regularized covariance from data using sklearn.
21232125

mne/decoding/csp.py

Lines changed: 8 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -18,6 +18,7 @@
1818
_validate_type,
1919
_verbose_safe_false,
2020
fill_doc,
21+
logger,
2122
pinv,
2223
)
2324
from .transformer import MNETransformerMixin
@@ -602,6 +603,12 @@ def _concat_cov(self, x_class, *, cov_kind, log_rank):
602603

603604
def _epoch_cov(self, x_class, *, cov_kind, log_rank):
604605
"""Mean of per-epoch covariances."""
606+
name = self.reg if isinstance(self.reg, str) else "empirical"
607+
name += " with shrinkage" if isinstance(self.reg, float) else ""
608+
logger.info(
609+
f"Estimating {cov_kind + (' ' if cov_kind else '')}"
610+
f"covariance (average over epochs; {name.upper()})"
611+
)
605612
cov = sum(
606613
_regularized_covariance(
607614
this_X,
@@ -612,6 +619,7 @@ def _epoch_cov(self, x_class, *, cov_kind, log_rank):
612619
cov_kind=cov_kind,
613620
log_rank=log_rank and ii == 0,
614621
log_ch_type="data",
622+
verbose=_verbose_safe_false(),
615623
)
616624
for ii, this_X in enumerate(x_class)
617625
)

0 commit comments

Comments
 (0)