【发布时间】:2021-09-09 18:48:48
【问题描述】:
使用 SciKit learn PolynomialFeatures 包时,可以执行以下操作:
- 拍摄特征:
x1、x2、x3 - 创建交互变量:
[x1, x2, x3, x1x2, x2x3, x1x3]使用PolynomialFeatures(interaction_only=True)
我的问题是我只想要x1 和所有其他术语之间的交互术语含义:
[x1, x2, x3, x1x2, x1x3],我不要x2x3。
可以这样做吗?
【问题讨论】:
-
为什么不通过切片获得输出数组的所需列?例如,如果您定义像
poly = PolynomialFeatures(include_bias=False, interaction_only=True)这样的对象并像y = poly.fit_transform(X)这样拟合变换,那么您只能通过y=y[:,:X.shape[1]*2-1]获得“X0”交互。
标签: python pandas numpy scikit-learn