diff --git a/R-package/R/xgb.plot.tree.R b/R-package/R/xgb.plot.tree.R index 63bebf6cf..2976f1b07 100644 --- a/R-package/R/xgb.plot.tree.R +++ b/R-package/R/xgb.plot.tree.R @@ -10,8 +10,8 @@ #' @param filename_dump the path to the text file storing the model. Model dump must include the gain per feature and per tree (parameter \code{with.stats = T} in function \code{xgb.dump}). Possible to provide a model directly (see \code{model} argument). #' @param model generated by the \code{xgb.train} function. Avoid the creation of a dump file. #' @param n_first_tree limit the plot to the n first trees. If \code{NULL}, all trees of the model are plotted. Performance can be low for huge models. -#' @param width the width of the diagram in pixels. -#' @param height the height of the diagram in pixels. +#' @param plot.width the width of the diagram in pixels. +#' @param plot.height the height of the diagram in pixels. #' #' @return A \code{DiagrammeR} of the model. #' @@ -43,7 +43,7 @@ #' xgb.plot.tree(agaricus.train$data@@Dimnames[[2]], model = bst) #' #' @export -xgb.plot.tree <- function(feature_names = NULL, filename_dump = NULL, model = NULL, n_first_tree = NULL, width = NULL, height = NULL){ +xgb.plot.tree <- function(feature_names = NULL, filename_dump = NULL, model = NULL, n_first_tree = NULL, plot.width = NULL, plot.height = NULL){ if (!class(model) %in% c("xgb.Booster", "NULL")) { stop("model: Has to be an object of class xgb.Booster model generaged by the xgb.train function.") @@ -87,7 +87,7 @@ xgb.plot.tree <- function(feature_names = NULL, filename_dump = NULL, model = NU edges_df = edges, graph_attrs = "rankdir = LR") - DiagrammeR::render_graph(graph, width = width, height = height) + DiagrammeR::render_graph(graph, width = plot.width, height = plot.height) } # Avoid error messages during CRAN check.