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: 一个整数,制定了要绘制的子数的编号。默认为 0rankdir: 一个字符串,它传递给graphviz的graph_attrax: 一个matplotlib Axes对象。特征重要性将绘制在它上面。 如果为None,则新建一个Axeskwargs: 关键字参数,用于传递给graphviz的graph_attr
返回ax (一个matplotlib Axes 对象)
学员评价