溫馨提示×

Scikit-learn中怎么使用特征選擇

小億
100
2024-05-10 17:05:53
欄目: 編程語言

在Scikit-learn中,可以使用特征選擇技術通過sklearn.feature_selection模塊中提供的方法來選擇最重要的特征。下面是一個簡單的例子來展示如何使用特征選擇:

from sklearn.datasets import load_iris
from sklearn.feature_selection import SelectKBest
from sklearn.feature_selection import chi2

# 加載數據集
iris = load_iris()
X, y = iris.data, iris.target

# 使用卡方檢驗選擇最重要的特征
k_best = SelectKBest(score_func=chi2, k=2)
X_new = k_best.fit_transform(X, y)

# 打印選擇的特征
print(X_new.shape)

在上面的例子中,我們使用了SelectKBest類來選擇最重要的2個特征,其中我們使用了卡方檢驗(chi2)作為評分函數。最后打印出選擇的特征的形狀。 此外,Scikit-learn還提供了其他的特征選擇方法,如SelectFromModelRFE等,具體可以查看官方文檔以了解更多信息。

0
亚洲午夜精品一区二区_中文无码日韩欧免_久久香蕉精品视频_欧美主播一区二区三区美女