使用随机森林计算特征重要性
from sklearn.ensemble import RandomForestRegressor
将特征和目标变量分开
X
y = endog
训练随机森林模型
rf = RandomForestRegressor()
rf.fit(X, y)
获取特征重要性
feature_importances = pd.Series(rf.feature_importances_, index=X.columns)
feature_importances.sort_values(ascending=False, inplace=True)
打印特征重要性
print(feature_importances)
版权声明:本文为red20200120原创文章,遵循 CC 4.0 BY-SA 版权协议,转载请附上原文出处链接和本声明。