sklearn.preprocessing.PolynomialFeatures

生成多项式和交互特征。

生成一个新的特征矩阵，该矩阵由度数小于或等于指定度数的特征的所有多项式组合组成。例如，如果输入样本是二维的且形式为 [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中进行特征预处理