4分钟
xgboost.tp_graphviz()
xgboost.tp_graphviz(): 转换指定的子树成一个graphviz 实例。
在IPython中,可以自动绘制graphviz 实例;否则你需要手动调用graphviz 对象的.render() 方法来绘制。
xgboost.to_graphviz(booster, fmap='', num_trees=0, rankdir='UT', yes_color='#0000FF',
no_color='#FF0000', **kwargs)参数:
yes_color: 一个字符串,给出了满足node condition的边的颜色no_color: 一个字符串,给出了不满足node condition的边的颜色- 其它参数参考
xgboost.plot_tree()
返回ax (一个matplotlib Axes 对象)
学员评价