diff --git a/R-package/NAMESPACE b/R-package/NAMESPACE index f3a7390b7..3fb05b7d8 100644 --- a/R-package/NAMESPACE +++ b/R-package/NAMESPACE @@ -21,10 +21,6 @@ exportMethods(predict) import(methods) importClassesFrom(Matrix,dgCMatrix) importClassesFrom(Matrix,dgeMatrix) -importFrom(DiagrammeR,create_edges) -importFrom(DiagrammeR,create_graph) -importFrom(DiagrammeR,create_nodes) -importFrom(DiagrammeR,render_graph) importFrom(Matrix,cBind) importFrom(Matrix,colSums) importFrom(Matrix,sparseVector) diff --git a/R-package/R/xgb.plot.tree.R b/R-package/R/xgb.plot.tree.R index 9977748db..475be7231 100644 --- a/R-package/R/xgb.plot.tree.R +++ b/R-package/R/xgb.plot.tree.R @@ -9,10 +9,6 @@ #' @importFrom data.table := #' @importFrom data.table copy #' @importFrom magrittr %>% -#' @importFrom DiagrammeR create_nodes -#' @importFrom DiagrammeR create_edges -#' @importFrom DiagrammeR create_graph -#' @importFrom DiagrammeR render_graph #' @param feature_names names of each feature as a character vector. Can be extracted from a sparse matrix (see example). If model dump already contains feature names, this argument should be \code{NULL}. #' @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.