Fix: Ensure power features have shape (64, 129) in BCIProcessor

VishwamAI · Oct 18, 2024 · 5dea08e · 5dea08e
1 parent a6abfb7
commit 5dea08e
Showing 1 changed file with 1 addition and 1 deletion.
diff --git a/NeuroFlex/bci_integration/bci_processing.py b/NeuroFlex/bci_integration/bci_processing.py
@@ -125,7 +125,7 @@ def extract_features(self, filtered_data: Dict[str, np.ndarray]) -> Dict[str, np
             elif psd.shape[1] < 129:
                 psd = np.pad(psd, ((0, 0), (0, 129 - psd.shape[1])))
 
-            features[f'{band}_power'] = psd.T  # Transpose to get shape (129, 64)
+            features[f'{band}_power'] = psd  # Shape is now (64, 129)
 
             # Apply wavelet transform
             coeffs = pywt.wavedec(data, 'db4', level=min(5, data.shape[-1] // 2), axis=-1)