Fix missing dependencies

This commit is contained in:
pommedeterresautee 2015-11-27 18:19:51 +01:00
parent 68b666d7e5
commit 5e9f4dc973
3 changed files with 5 additions and 4 deletions

View File

@ -37,6 +37,7 @@ importFrom(data.table,setnames)
importFrom(magrittr,"%>%")
importFrom(magrittr,add)
importFrom(magrittr,not)
importFrom(stringr,str_detect)
importFrom(stringr,str_extract)
importFrom(stringr,str_extract_all)
importFrom(stringr,str_match)

View File

@ -12,7 +12,6 @@
#' @importFrom magrittr add
#' @importFrom stringr str_extract
#' @importFrom stringr str_split
#' @importFrom stringr str_extract
#' @importFrom stringr str_trim
#' @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}).

View File

@ -7,6 +7,8 @@
#' @importFrom data.table setnames
#' @importFrom data.table :=
#' @importFrom magrittr %>%
#' @importFrom stringr str_detect
#' @importFrom stringr str_extract
#'
#' @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}).
#' @param model dump generated by the \code{xgb.train} function. Avoid the creation of a dump file.
@ -83,9 +85,8 @@ xgb.plot.multi.trees <- function(model, names, features.keep = 5, plot.width = N
label = nodes.dt[,Text],
style = "filled",
color = "DimGray",
fillcolor= "Blue",
fillcolor= "Beige",
shape = "oval",
#data = allTrees[,Feature]
fontname = "Helvetica"
)