getinfo.Rd 1.18 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
}
\value{
20
21
info data

Guolin Ke's avatar
Guolin Ke committed
22
23
24
info data
}
\description{
25
Get one attribute of a \code{lgb.Dataset}
Guolin Ke's avatar
Guolin Ke committed
26
27
28
29
30
31
32
}
\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
33
34
    \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
35
36
37
}
}
\examples{
38
\donttest{
Guolin Ke's avatar
Guolin Ke committed
39
40
41
42
43
44
45
46
47
48
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))
49
}
Guolin Ke's avatar
Guolin Ke committed
50
}