28 lines
675 B
R
28 lines
675 B
R
% Generated by roxygen2 (4.0.1): do not edit by hand
|
|
\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 binary file.}
|
|
}
|
|
\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, nround = 2,objective = "binary:logistic")
|
|
xgb.save(bst, 'xgb.model')
|
|
bst <- xgb.load('xgb.model')
|
|
pred <- predict(bst, test$data)
|
|
}
|
|
|