Hendrik/xgboost
1
0
Fork 0
Code Issues Pull Requests Packages Projects Releases Wiki Activity

better doc of dump function

Browse Source
This commit is contained in:
El Potaeto 2015-01-09 11:14:08 +01:00
parent 9d6eecf34e
commit 31d0e8f65d
2 changed files with 13 additions and 5 deletions
Show Stats Download Patch File Download Diff File Expand all files Collapse all files

10
R-package/R/xgb.dump.R
View File

@ -6,7 +6,7 @@
#' @importFrom stringr str_split #' @importFrom stringr str_split
#' @importFrom stringr str_replace #' @importFrom stringr str_replace
#' @param model the model object. #' @param model the model object.
#' @param fname the name of the text file where to save the model. If not provided or set to \code{NULL} the function will return the model as a \code{character} vector. #' @param fname the name of the text file where to save the model text dump. If not provided or set to \code{NULL} the function will return the model as a \code{character} vector.
#' @param fmap feature map file representing the type of feature. #' @param fmap feature map file representing the type of feature.
#' Detailed description could be found at #' Detailed description could be found at
#' \url{https://github.com/tqchen/xgboost/wiki/Binary-Classification#dump-model}. #' \url{https://github.com/tqchen/xgboost/wiki/Binary-Classification#dump-model}.
@ -28,15 +28,19 @@
#' test <- agaricus.test #' test <- agaricus.test
#' bst <- xgboost(data = train$data, label = train$label, max.depth = 2, #' bst <- xgboost(data = train$data, label = train$label, max.depth = 2,
#' eta = 1, nround = 2,objective = "binary:logistic") #' eta = 1, nround = 2,objective = "binary:logistic")
#' # save the model in file 'xgb.model.dump'
#' xgb.dump(bst, 'xgb.model.dump') #' xgb.dump(bst, 'xgb.model.dump')
#'
#' # print the model without saving it to a file
#' print(xgb.dump(bst))
#' @export #' @export
#' #'
xgb.dump <- function(model, fname = NULL, fmap = "", with.stats=FALSE) { xgb.dump <- function(model = NULL, fname = NULL, fmap = "", with.stats=FALSE) {
if (class(model) != "xgb.Booster") { if (class(model) != "xgb.Booster") {
stop("xgb.dump: first argument must be type xgb.Booster") stop("xgb.dump: first argument must be type xgb.Booster")
} }
if (!class(fname) %in% c("character", "NULL")) { if (!class(fname) %in% c("character", "NULL")) {
stop("xgb.dump: second argument must be type character if provided") stop("xgb.dump: second argument must be type character when provided")
} }
result <- .Call("XGBoosterDumpModel_R", model, fmap, as.integer(with.stats), PACKAGE = "xgboost") result <- .Call("XGBoosterDumpModel_R", model, fmap, as.integer(with.stats), PACKAGE = "xgboost")

8
R-package/man/xgb.dump.Rd
View File

@ -4,12 +4,12 @@
\alias{xgb.dump} \alias{xgb.dump}
\title{Save xgboost model to text file} \title{Save xgboost model to text file}
\usage{ \usage{
xgb.dump(model, fname = NULL, fmap = "", with.stats = FALSE) xgb.dump(model = NULL, fname = NULL, fmap = "", with.stats = FALSE)
} }
\arguments{ \arguments{
\item{model}{the model object.} \item{model}{the model object.}
\item{fname}{the name of the text file where to save the model. If not provided or set to \code{NULL} the function will return the model as a \code{character} vector.} \item{fname}{the name of the text file where to save the model text dump. If not provided or set to \code{NULL} the function will return the model as a \code{character} vector.}
\item{fmap}{feature map file representing the type of feature. \item{fmap}{feature map file representing the type of feature.
Detailed description could be found at Detailed description could be found at
@ -36,6 +36,10 @@ train <- agaricus.train
test <- agaricus.test test <- agaricus.test
bst <- xgboost(data = train$data, label = train$label, max.depth = 2, bst <- xgboost(data = train$data, label = train$label, max.depth = 2,
eta = 1, nround = 2,objective = "binary:logistic") eta = 1, nround = 2,objective = "binary:logistic")
# save the model in file 'xgb.model.dump'
xgb.dump(bst, 'xgb.model.dump') xgb.dump(bst, 'xgb.model.dump')
# print the model without saving it to a file
print(xgb.dump(bst))
} }