机器学习–特征

  • Post author:
  • Post category:其他




使用随机森林计算特征重要性

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 版权协议,转载请附上原文出处链接和本声明。