error early if missing DiagrammeR (#8037)

This commit is contained in:
Michael Chirico
2022-07-02 04:37:53 -07:00
committed by GitHub
parent bc4f802b17
commit 3af02584c1

View File

@@ -62,6 +62,9 @@
#' @export
xgb.plot.multi.trees <- function(model, feature_names = NULL, features_keep = 5, plot_width = NULL, plot_height = NULL,
render = TRUE, ...){
if (!requireNamespace("DiagrammeR", quietly = TRUE)) {
stop("DiagrammeR is required for xgb.plot.multi.trees")
}
check.deprecation(...)
tree.matrix <- xgb.model.dt.tree(feature_names = feature_names, model = model)