From bfb221a05a874e12fa6929daaac02690c4e1f33d Mon Sep 17 00:00:00 2001 From: Anuja Negi Date: Thu, 8 Feb 2024 12:30:54 +0100 Subject: [PATCH] removed shape change in free orient for eloreta --- bsi_zoo/estimators.py | 5 +++-- 1 file changed, 3 insertions(+), 2 deletions(-) diff --git a/bsi_zoo/estimators.py b/bsi_zoo/estimators.py index c64bbf8..8cd5d80 100644 --- a/bsi_zoo/estimators.py +++ b/bsi_zoo/estimators.py @@ -412,6 +412,7 @@ def __init__(self, solver, alpha, cov_type, cov, n_orient, extra_params={}): self.cov_type = cov_type self.n_orient = n_orient self.extra_params = extra_params + self.classes_ = np.array([0, 0]) def fit(self, L, y): self.L_ = L @@ -716,8 +717,8 @@ def eloreta(L, y, cov=1, alpha=1 / 9, n_orient=1): K = _compute_eloreta_kernel(L, lambda2=alpha, n_orient=n_orient, whitener=whitener) x = K @ y # get the source time courses with simple dot product - if n_orient > 1: - x = x.reshape((-1, n_orient, x.shape[1])) + # if n_orient > 1: + # x = x.reshape((-1, n_orient, x.shape[1])) return x