setinfo.Rd 1.21 KB
Newer Older
Guolin Ke's avatar
Guolin Ke committed
1
2
3
4
5
% Generated by roxygen2: do not edit by hand
% Please edit documentation in R/lgb.Dataset.R
\name{setinfo}
\alias{setinfo}
\alias{setinfo.lgb.Dataset}
Nikita Titov's avatar
Nikita Titov committed
6
\title{Set information of an \code{lgb.Dataset} object}
Guolin Ke's avatar
Guolin Ke committed
7
8
9
10
11
12
\usage{
setinfo(dataset, ...)

\method{setinfo}{lgb.Dataset}(dataset, name, info, ...)
}
\arguments{
Nikita Titov's avatar
Nikita Titov committed
13
\item{dataset}{Object of class \code{lgb.Dataset}}
Guolin Ke's avatar
Guolin Ke committed
14
15
16
17

\item{name}{the name of the field to get}

\item{info}{the specific field of information to set}
18
19

\item{...}{other parameters}
Guolin Ke's avatar
Guolin Ke committed
20
21
22
23
24
}
\value{
passed object
}
\description{
Nikita Titov's avatar
Nikita Titov committed
25
Set information of an \code{lgb.Dataset} object
Guolin Ke's avatar
Guolin Ke committed
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
}
\details{
The \code{name} field can be one of the following:

\itemize{
    \item \code{label}: label lightgbm learn from ;
    \item \code{weight}: to do a weight rescale ;
    \item \code{init_score}: initial score is the base prediction lightgbm will boost from ;
    \item \code{group}.
}
}
\examples{
library(lightgbm)
data(agaricus.train, package = "lightgbm")
train <- agaricus.train
dtrain <- lgb.Dataset(train$data, label = train$label)
lgb.Dataset.construct(dtrain)

labels <- lightgbm::getinfo(dtrain, "label")
lightgbm::setinfo(dtrain, "label", 1 - labels)

labels2 <- lightgbm::getinfo(dtrain, "label")
stopifnot(all.equal(labels2, 1 - labels))

}