生成多项式和交互特征。
生成一个新的特征矩阵,该矩阵由度数小于或等于指定度数的特征的所有多项式组合组成。例如,如果输入样本是二维的且形式为 [a, b],则 2 次多项式特征为 [1, a, b, a^2, ab, b^2]。
文档链接
>>> import numpy as np
>>> from sklearn.preprocessing import PolynomialFeatures
>>> X = np.arange(6).reshape(3, 2)
>>> X
array([[0, 1],
[2, 3],
[4, 5]])
>>> poly = PolynomialFeatures(2)
>>> poly.fit_transform(X)
array([[ 1., 0., 1., 0., 0., 1.],
[ 1., 2., 3., 4., 6., 9.],
[ 1., 4., 5., 16., 20., 25.]])
>>> poly = PolynomialFeatures(interaction_only=True)
>>> poly.fit_transform(X)
array([[ 1., 0., 1., 0.],
[ 1., 2., 3., 6.],
[ 1., 4., 5., 20.]])
PolynomialFeatures可以用户来提取特征,在wide_and_deep中进行特征预处理