首页 > 解决方案 > 使用 StandardScaler 使用特定功能缩放具有不同形状的数组

问题描述

我有一个具有以下形状的数据集:

X =(样本,4)

y =(样本,)

我正在使用来自 sckikit-learn 的 StandardScaler 来规范化 X 的数据

我想用 X 数据集的四列之一中使用的相同特征来缩放 y 数据集,例如缩放 y 的所有值,类似于 X 的第二列。

有没有办法做到这一点?

标签: pythonnumpymachine-learningscikit-learnnormalization

解决方案


假设 X 中的列要在此基础上缩放 y,命名为“A”。你可以这样做:

from sklearn.preprocessing import StandardScaler
scaler = StandardScaler()
scaler.fit(X['A'])
y = scaler.tansform(y)

推荐阅读