FRE 和 随机森林 特征重要性

#使用ref进行特征选择
from sklearn.linear_model import LinearRegression, Ridge
from sklearn import svm
from sklearn.feature_selection import RFE
lr = LinearRegression()
rfe = RFE(lr, n_features_to_select=5)
rfe.fit(train_X,label_1)
print ("Features sorted by their rank:")
print( sorted(zip(map(lambda x: round(x, 4), rfe.ranking_), feature))  )#使用随机森林进行特征选择
from sklearn.ensemble import RandomForestRegressor
import numpy as np
rf = RandomForestRegressor()
rf.fit(train_X,label_1)
print ("Features sorted by their score:")
print (sorted(zip(map(lambda x: round(x, 4), rf.feature_importances_), feature),reverse=True))


本文来自互联网用户投稿,文章观点仅代表作者本人,不代表本站立场,不承担相关法律责任。如若转载,请注明出处。 如若内容造成侵权/违法违规/事实不符,请点击【内容举报】进行投诉反馈!

相关文章

立即
投稿

微信公众账号

微信扫一扫加关注

返回
顶部