rfe特征选择

from sklearn.datasets import make_friedman1
from sklearn.feature_selection import RFECV
from sklearn.svm import SVR
import pandas
from sklearn.preprocessing import scale

data = pandas.read_csv("iris.csv")
print(data)
print(data.shape)
X = data.iloc[:,0:-1]
#def z_score(x):
X = scale(X)
y = data.iloc[:,-1]
print(X.shape)
estimator = SVR(kernel="linear")
selector = RFECV(estimator, step=1, cv=5)
selector = selector.fit(X, y)
print(selector.support_)
print(selector.ranking_)

猜你喜欢

转载自blog.csdn.net/zanlinux/article/details/108664075