29 lines
728 B
R
29 lines
728 B
R
% Generated by roxygen2: do not edit by hand
|
|
% Please edit documentation in R/xgb.save.R
|
|
\name{xgb.save}
|
|
\alias{xgb.save}
|
|
\title{Save xgboost model to binary file}
|
|
\usage{
|
|
xgb.save(model, fname)
|
|
}
|
|
\arguments{
|
|
\item{model}{the model object.}
|
|
|
|
\item{fname}{the name of the file to write.}
|
|
}
|
|
\description{
|
|
Save xgboost model from xgboost or xgb.train
|
|
}
|
|
\examples{
|
|
data(agaricus.train, package='xgboost')
|
|
data(agaricus.test, package='xgboost')
|
|
train <- agaricus.train
|
|
test <- agaricus.test
|
|
bst <- xgboost(data = train$data, label = train$label, max_depth = 2,
|
|
eta = 1, nthread = 2, nrounds = 2,objective = "binary:logistic")
|
|
xgb.save(bst, 'xgb.model')
|
|
bst <- xgb.load('xgb.model')
|
|
pred <- predict(bst, test$data)
|
|
}
|
|
|