随机森林重要性排序代码
时间: 2023-10-31 11:04:36 浏览: 121
基于随机森林RF的回归预测,随机森林RF重要性排序,多变量输入模型 运行环境为matlab2018,程序内注释详细,直接替换
以下是使用Python中sklearn库实现随机森林特征重要性排序的代码:
```python
from sklearn.ensemble import RandomForestRegressor
# 创建随机森林回归模型
rf = RandomForestRegressor(n_estimators=1000, random_state=42)
# 训练模型
rf.fit(X_train, y_train)
# 特征重要性排序
importances = list(rf.feature_importances_)
feature_importances = [(feature, round(importance, 2)) for feature, importance in zip(feature_list, importances)]
feature_importances = sorted(feature_importances, key = lambda x: x[1], reverse = True)
for pair in feature_importances:
print('特征:{:20} 重要性:{}'.format(*pair))
```
其中,X_train和y_train是训练集的特征和标签,feature_list是特征名称列表,可以根据具体情况进行修改。此代码将输出每个特征的重要性排名。
阅读全文