getinfo.Rd 1.16 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{getinfo}
\alias{getinfo}
\alias{getinfo.lgb.Dataset}
Nikita Titov's avatar
Nikita Titov committed
6
\title{Get information of an \code{lgb.Dataset} object}
Guolin Ke's avatar
Guolin Ke committed
7
8
9
10
11
12
13
14
\usage{
getinfo(dataset, ...)

\method{getinfo}{lgb.Dataset}(dataset, name, ...)
}
\arguments{
\item{dataset}{Object of class \code{lgb.Dataset}}

15
\item{...}{other parameters}
16
17

\item{name}{the name of the information field to get (see details)}
Guolin Ke's avatar
Guolin Ke committed
18
19
20
21
22
}
\value{
info data
}
\description{
23
Get one attribute of a \code{lgb.Dataset}
Guolin Ke's avatar
Guolin Ke committed
24
25
26
27
28
29
30
}
\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 ;
Nikita Titov's avatar
Nikita Titov committed
31
32
    \item \code{group}: group size ;
    \item \code{init_score}: initial score is the base prediction lightgbm will boost from.
Guolin Ke's avatar
Guolin Ke committed
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
}
}
\examples{
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(labels2 == 1 - labels))

}