使用随机森林计算特征重要性
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)