python - 如何标准缩放 3D 矩阵？

Question

我正在处理信号分类问题，想先缩放数据集矩阵，但我的数据是 3D 格式(批处理、长度、 channel )。
我尝试使用 Scikit-learn 标准缩放器:

from sklearn.preprocessing import StandardScaler
sc = StandardScaler()
X_train = sc.fit_transform(X_train)
X_test = sc.transform(X_test)

但是我收到了这个错误信息:

Found array with dim 3. StandardScaler expected <= 2

我认为一种解决方案是将矩阵按多个 2D 矩阵中的每个 channel 拆分，分别缩放它们，然后以 3D 格式放回原处，但我想知道是否有更好的解决方案。
非常感谢。

Answer 1

只有3行代码...

scaler = StandardScaler()
X_train = scaler.fit_transform(X_train.reshape(-1, X_train.shape[-1])).reshape(X_train.shape)
X_test = scaler.transform(X_test.reshape(-1, X_test.shape[-1])).reshape(X_test.shape)