43 lines
1.1 KiB
R
43 lines
1.1 KiB
R
% Generated by roxygen2: do not edit by hand
|
|
% Please edit documentation in R/xgb.DMatrix.R
|
|
\name{setinfo}
|
|
\alias{setinfo}
|
|
\alias{setinfo.xgb.DMatrix}
|
|
\title{Set information of an xgb.DMatrix object}
|
|
\usage{
|
|
setinfo(object, ...)
|
|
|
|
\method{setinfo}{xgb.DMatrix}(object, name, info, ...)
|
|
}
|
|
\arguments{
|
|
\item{object}{Object of class "xgb.DMatrix"}
|
|
|
|
\item{...}{other parameters}
|
|
|
|
\item{name}{the name of the field to get}
|
|
|
|
\item{info}{the specific field of information to set}
|
|
}
|
|
\description{
|
|
Set information of an xgb.DMatrix object
|
|
}
|
|
\details{
|
|
The \code{name} field can be one of the following:
|
|
|
|
\itemize{
|
|
\item \code{label}: label XGBoost learn from ;
|
|
\item \code{weight}: to do a weight rescale ;
|
|
\item \code{base_margin}: base margin is the base prediction XGBoost will boost from ;
|
|
\item \code{group}: number of rows in each group (to use with \code{rank:pairwise} objective).
|
|
}
|
|
}
|
|
\examples{
|
|
data(agaricus.train, package='xgboost')
|
|
dtrain <- with(agaricus.train, xgb.DMatrix(data, label = label))
|
|
|
|
labels <- getinfo(dtrain, 'label')
|
|
setinfo(dtrain, 'label', 1-labels)
|
|
labels2 <- getinfo(dtrain, 'label')
|
|
stopifnot(all.equal(labels2, 1-labels))
|
|
}
|