6分钟
xgboost.plot_importance()
xgboost.plot_importance():绘制特征重要性
xgboost.plot_importance(booster, ax=None, height=0.2, xlim=None, ylim=None,
title='Feature importance', xlabel='F score', ylabel='Features',
importance_type='weight', max_num_features=None, grid=True,
show_values=True, **kwargs)参数:
booster: 一个Booster对象, 一个XGBModel对象,或者由Booster.get_fscore()返回的字典ax: 一个matplotlib Axes对象。特征重要性将绘制在它上面。 如果为None,则新建一个Axesgrid: 一个布尔值。如果为True,则开启axes gridimportance_type: 一个字符串,指定了特征重要性的类别。参考Booster.get_fscore()max_num_features: 一个整数,指定展示的特征的最大数量。如果为None,则展示所有的特征height: 一个浮点数,指定bar的高度。它传递给ax.barh()xlim: 一个元组,传递给axes.xlim()ylim: 一个元组,传递给axes.ylim()title: 一个字符串,设置Axes的标题。默认为"Feature importance"。 如果为None,则没有标题xlabel: 一个字符串,设置Axes的X轴标题。默认为"F score"。 如果为None,则X轴没有标题ylabel:一个字符串,设置Axes的Y轴标题。默认为"Features"。 如果为None,则Y轴没有标题show_values: 一个布尔值。如果为True,则在绘图上展示具体的值。kwargs: 关键字参数,用于传递给ax.barh()
返回ax (一个matplotlib Axes 对象)
学员评价