课程评价 (0)

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

学员评价

暂无精选评价
4分钟

xgboost.plot_tree()

xgboost.plot_tree(): 绘制指定的子树。

xgboost.plot_tree(booster, fmap='', num_trees=0, rankdir='UT', ax=None, **kwargs)

参数:

  • booster: 一个Booster对象, 一个 XGBModel 对象
  • fmap: 一个字符串,给出了feature map 文件的文件名
  • num_trees: 一个整数,制定了要绘制的子数的编号。默认为 0
  • rankdir: 一个字符串,它传递给graphvizgraph_attr
  • ax: 一个matplotlib Axes 对象。特征重要性将绘制在它上面。 如果为None,则新建一个Axes
  • kwargs: 关键字参数,用于传递给graphvizgraph_attr

返回ax (一个matplotlib Axes 对象)