如何使用scikit-learn仅删除多项式回归中的交互项?

Har*_*nna 3 machine-learning python-2.7 scikit-learn non-linear-regression

我正在使用scikit-learn 运行多项式回归。我有大量变量(准确地说是23个),我试图使用2级多项式回归进行回归。

interact_only = True,仅保留交互项,例如X 1 * Y 1,X 2 * Y 2等等。

我只需要其他术语,即X 1,X 1 2,Y 1,Y 1 2,依此类推。

有功能可以得到这个吗?

lej*_*lot 5

没有这样的功能,因为转换很容易用numpy本身表示。

X = ... 
new_X = np.hstack((X, X**2))
Run Code Online (Sandbox Code Playgroud)

类似地,如果您想将所有内容加起来 k

new_X = np.hstack((X**(i+1) for i in range(k)))
Run Code Online (Sandbox Code Playgroud)

  • 我会给你一个更好的:`np.power(x, np.arange(k))` (2认同)