📜  随机福雷斯特绘制特征重要性函数 - Python 代码示例

📅  最后修改于: 2022-03-11 14:45:19.397000             🧑  作者: Mango

代码示例1
def plot_feature_importances(model):
    n_features = data_train.shape[1]
    plt.figure(figsize=(20,20))
    plt.barh(range(n_features), model.feature_importances_, align='center') 
    plt.yticks(np.arange(n_features), data_train.columns.values) 
    plt.xlabel('Feature importance')
    plt.ylabel('Feature')