Skip to contents

Plot XGBoost tree visualization

Source: R/supervised-xgboost.R
tl_plot_xgboost_tree.Rd

Plot XGBoost tree visualization

Usage

tl_plot_xgboost_tree(model, tree_index = 0, ...)

Arguments

model

A tidylearn XGBoost model object

tree_index

Index of the tree to plot (default: 0, first tree)

...

Additional arguments

Value

The return value of xgb.plot.tree, a tree diagram rendered via the DiagrammeR package.