Sklearn-preprocessing.PolynomialFeatures

sklearn /

python /

数据分析


在建模过程中多次用到过sklearn.preprocessing.PolynomialFeatures,可以理解为专门生成多项式特征,并且多项式包含的是相互影响的特征集,比如:一个输入样本是2维的。形式如[a,b] ,则二阶多项式的特征集如下[1,a,b,a^2,ab,b^2]。

官网文档:http://scikit-learn.org/stable/modules/generated/sklearn.preprocessing.PolynomialFeatures.html

参数:

degree : integer,多项式阶数,默认为2;

interaction_only : boolean, default = False,如果值为true(默认是false),则会产生相互影响的特征集;

include_bias : boolean,是否包含偏差列。

示例:

[python]view plaincopy

>>> X = np.arange(6).reshape(3, 2)  

>>> X  

array([[0, 1],  

[2, 3],  

[4, 5]])  

>>> poly = PolynomialFeatures(2) #设置多项式阶数为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)#默认的阶数是2,同时设置交互关系为true  

>>> poly.fit_transform(X)  

array([[1.,   0.,   1.,   0.],  

[1.,   2.,   3.,   6.],  

[1.,   4.,   5.,  20.]])  

上面的数组中,每一行是一个list。比如[0,1] 类似与上面的[a,b]。好的现在它的多项式输出矩阵就是[1,a,b,a^2,ab,b^2]。所以就是下面对应的[1,0,1,0,0,1]。现在将interaction_only=True。这时就是只找交互作用的多项式输出矩阵。例如[a,b]的多项式交互式输出[1,a,b,ab]。不存在自己与自己交互的情况如;a^2或者a*b^2之类的。

你可能感兴趣的:(Sklearn-preprocessing.PolynomialFeatures)