1. 程式人生 > >特徵選擇之方差選擇法VarianceThreshold

特徵選擇之方差選擇法VarianceThreshold

VarianceThreshold

#方差選擇法 #使用方差選擇法,先要計算各個特徵的方差,然後根據閾值,選擇方差大於閾值的特徵。使用feature_selection庫的VarianceThreshold類來選擇特徵的程式碼如下: from sklearn.feature_selection import VarianceThreshold #方差選擇法,返回值為特徵選擇後的資料 #引數threshold為方差的閾值 from sklearn.datasets import load_iris iris = load_iris() #print(VarianceThreshold(threshold=3).fit_transform(iris.data)) print(iris.data[0:5]) selector = VarianceThreshold(threshold=3).fit(iris.data, iris.target) data = selector.transform(iris.data) print(data[0:5]) print(selector.variances_)