课程评价 (0)

请对课程作出评价:
0/300

学员评价

暂无精选评价
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,则新建一个Axes
  • grid: 一个布尔值。如果为True,则开启axes grid
  • importance_type: 一个字符串,指定了特征重要性的类别。参考Booster.get_fscore()
  • max_num_features: 一个整数,指定展示的特征的最大数量。如果为None,则展示所有的特征
  • height: 一个浮点数,指定bar 的高度。它传递给ax.barh()
  • xlim: 一个元组,传递给 axes.xlim()
  • ylim: 一个元组,传递给 axes.ylim()
  • title: 一个字符串,设置Axes 的标题。默认为"Feature importance"。 如果为None,则没有标题
  • xlabel: 一个字符串,设置AxesX 轴标题。默认为"F score"。 如果为None,则X 轴没有标题
  • ylabel:一个字符串,设置AxesY 轴标题。默认为"Features"。 如果为None,则Y 轴没有标题
  • show_values: 一个布尔值。如果为True,则在绘图上展示具体的值。
  • kwargs: 关键字参数,用于传递给ax.barh()

返回ax (一个matplotlib Axes 对象)