生成多項式和交互特征。

生成一個新的特征矩陣，該矩陣由度數小于或等于指定度數的特征的所有多項式組合組成。例如，如果輸入樣本是二維的且形式為 [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中進行特征預處理

總結

以上是生活随笔為你收集整理的sklearn.preprocessing.PolynomialFeatures的全部內容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。