some fixes for Travis #Rstat

R-package/R/xgb.plot.multi.trees.R

@@ -11,6 +11,7 @@
 #' @importFrom stringr str_extract
 #' 
 #' @param model dump generated by the \code{xgb.train} function. Avoid the creation of a dump file.
+#' @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 features.keep number of features to keep in each position of the multi trees.
 #' @param plot.width width in pixels of the graph to produce
 #' @param plot.height height in pixels of the graph to produce
@@ -45,12 +46,12 @@
 #'                  eta = 1, nthread = 2, nround = 30, objective = "binary:logistic",
 #'                  min_child_weight = 50)
 #'
-#' p <- xgb.plot.multi.trees(model = bst, names = agaricus.train$data@Dimnames[[2]], 3)
+#' p <- xgb.plot.multi.trees(model = bst, feature_names = agaricus.train$data@Dimnames[[2]], features.keep = 3)
 #' print(p)
 #'
 #' @export
-xgb.plot.multi.trees <- function(model, names, features.keep = 5, plot.width = NULL, plot.height = NULL){
-  tree.matrix <- xgb.model.dt.tree(names, model = model)
+xgb.plot.multi.trees <- function(model, feature_names = NULL, features.keep = 5, plot.width = NULL, plot.height = NULL){
+  tree.matrix <- xgb.model.dt.tree(feature_names = feature_names, model = model)
   
   # first number of the path represents the tree, then the following numbers are related to the path to follow
   # root init