Skip to content
GitLab
Menu
Projects
Groups
Snippets
Loading...
Help
Help
Support
Community forum
Keyboard shortcuts
?
Submit feedback
Contribute to GitLab
Sign in / Register
Toggle navigation
Menu
Open sidebar
tianlh
LightGBM-DCU
Commits
7dcbb8cd
Unverified
Commit
7dcbb8cd
authored
Jul 19, 2023
by
James Lamb
Committed by
GitHub
Jul 19, 2023
Browse files
[R-package] limit number of threads used in tests and examples (fixes #5987) (#5988)
parent
7d4d8975
Changes
40
Hide whitespace changes
Inline
Side-by-side
Showing
20 changed files
with
38 additions
and
9 deletions
+38
-9
R-package/R/callback.R
R-package/R/callback.R
+6
-6
R-package/R/lgb.Booster.R
R-package/R/lgb.Booster.R
+9
-1
R-package/R/lgb.cv.R
R-package/R/lgb.cv.R
+1
-0
R-package/R/lgb.importance.R
R-package/R/lgb.importance.R
+1
-0
R-package/R/lgb.interprete.R
R-package/R/lgb.interprete.R
+1
-0
R-package/R/lgb.model.dt.tree.R
R-package/R/lgb.model.dt.tree.R
+1
-0
R-package/R/lgb.plot.importance.R
R-package/R/lgb.plot.importance.R
+1
-0
R-package/R/lgb.plot.interpretation.R
R-package/R/lgb.plot.interpretation.R
+1
-0
R-package/R/lgb.restore_handle.R
R-package/R/lgb.restore_handle.R
+3
-1
R-package/R/lgb.train.R
R-package/R/lgb.train.R
+1
-0
R-package/R/readRDS.lgb.Booster.R
R-package/R/readRDS.lgb.Booster.R
+1
-0
R-package/R/saveRDS.lgb.Booster.R
R-package/R/saveRDS.lgb.Booster.R
+1
-0
R-package/man/lgb.configure_fast_predict.Rd
R-package/man/lgb.configure_fast_predict.Rd
+4
-1
R-package/man/lgb.cv.Rd
R-package/man/lgb.cv.Rd
+1
-0
R-package/man/lgb.dump.Rd
R-package/man/lgb.dump.Rd
+1
-0
R-package/man/lgb.get.eval.result.Rd
R-package/man/lgb.get.eval.result.Rd
+1
-0
R-package/man/lgb.importance.Rd
R-package/man/lgb.importance.Rd
+1
-0
R-package/man/lgb.interprete.Rd
R-package/man/lgb.interprete.Rd
+1
-0
R-package/man/lgb.load.Rd
R-package/man/lgb.load.Rd
+1
-0
R-package/man/lgb.model.dt.tree.Rd
R-package/man/lgb.model.dt.tree.Rd
+1
-0
No files found.
R-package/R/callback.R
View file @
7dcbb8cd
...
@@ -24,7 +24,7 @@ CB_ENV <- R6::R6Class(
...
@@ -24,7 +24,7 @@ CB_ENV <- R6::R6Class(
)
)
# Format the evaluation metric string
# Format the evaluation metric string
format
.
eval
.
string
<-
function
(
eval_res
,
eval_err
)
{
.
format
_
eval
_
string
<-
function
(
eval_res
,
eval_err
)
{
# Check for empty evaluation string
# Check for empty evaluation string
if
(
is.null
(
eval_res
)
||
length
(
eval_res
)
==
0L
)
{
if
(
is.null
(
eval_res
)
||
length
(
eval_res
)
==
0L
)
{
...
@@ -40,7 +40,7 @@ format.eval.string <- function(eval_res, eval_err) {
...
@@ -40,7 +40,7 @@ format.eval.string <- function(eval_res, eval_err) {
}
}
merge
.
eval
.
string
<-
function
(
env
)
{
.
merge
_
eval
_
string
<-
function
(
env
)
{
# Check length of evaluation list
# Check length of evaluation list
if
(
length
(
env
$
eval_list
)
<=
0L
)
{
if
(
length
(
env
$
eval_list
)
<=
0L
)
{
...
@@ -63,7 +63,7 @@ merge.eval.string <- function(env) {
...
@@ -63,7 +63,7 @@ merge.eval.string <- function(env) {
}
}
# Set error message
# Set error message
msg
<-
c
(
msg
,
format
.
eval
.
string
(
eval_res
=
env
$
eval_list
[[
j
]],
eval_err
=
eval_err
))
msg
<-
c
(
msg
,
.
format
_
eval
_
string
(
eval_res
=
env
$
eval_list
[[
j
]],
eval_err
=
eval_err
))
}
}
...
@@ -86,11 +86,11 @@ cb_print_evaluation <- function(period) {
...
@@ -86,11 +86,11 @@ cb_print_evaluation <- function(period) {
if
((
i
-
1L
)
%%
period
==
0L
||
is.element
(
i
,
c
(
env
$
begin_iteration
,
env
$
end_iteration
)))
{
if
((
i
-
1L
)
%%
period
==
0L
||
is.element
(
i
,
c
(
env
$
begin_iteration
,
env
$
end_iteration
)))
{
# Merge evaluation string
# Merge evaluation string
msg
<-
merge
.
eval
.
string
(
env
=
env
)
msg
<-
.
merge
_
eval
_
string
(
env
=
env
)
# Check if message is existing
# Check if message is existing
if
(
nchar
(
msg
)
>
0L
)
{
if
(
nchar
(
msg
)
>
0L
)
{
print
(
merge
.
eval
.
string
(
env
=
env
))
print
(
.
merge
_
eval
_
string
(
env
=
env
))
}
}
}
}
...
@@ -270,7 +270,7 @@ cb_early_stop <- function(stopping_rounds, first_metric_only, verbose) {
...
@@ -270,7 +270,7 @@ cb_early_stop <- function(stopping_rounds, first_metric_only, verbose) {
# Prepare to print if verbose
# Prepare to print if verbose
if
(
verbose
)
{
if
(
verbose
)
{
best_msg
[[
i
]]
<<-
as.character
(
merge
.
eval
.
string
(
env
=
env
))
best_msg
[[
i
]]
<<-
as.character
(
.
merge
_
eval
_
string
(
env
=
env
))
}
}
}
else
{
}
else
{
...
...
R-package/R/lgb.Booster.R
View file @
7dcbb8cd
...
@@ -928,6 +928,7 @@ NULL
...
@@ -928,6 +928,7 @@ NULL
#' , metric = "l2"
#' , metric = "l2"
#' , min_data = 1L
#' , min_data = 1L
#' , learning_rate = 1.0
#' , learning_rate = 1.0
#' , num_threads = 2L
#' )
#' )
#' valids <- list(test = dtest)
#' valids <- list(test = dtest)
#' model <- lgb.train(
#' model <- lgb.train(
...
@@ -1086,7 +1087,10 @@ predict.lgb.Booster <- function(object,
...
@@ -1086,7 +1087,10 @@ predict.lgb.Booster <- function(object,
#' X <- as.matrix(mtcars[, -1L])
#' X <- as.matrix(mtcars[, -1L])
#' y <- mtcars[, 1L]
#' y <- mtcars[, 1L]
#' dtrain <- lgb.Dataset(X, label = y, params = list(max_bin = 5L))
#' dtrain <- lgb.Dataset(X, label = y, params = list(max_bin = 5L))
#' params <- list(min_data_in_leaf = 2L)
#' params <- list(
#' min_data_in_leaf = 2L
#' , num_threads = 2L
#' )
#' model <- lgb.train(
#' model <- lgb.train(
#' params = params
#' params = params
#' , data = dtrain
#' , data = dtrain
...
@@ -1231,6 +1235,7 @@ summary.lgb.Booster <- function(object, ...) {
...
@@ -1231,6 +1235,7 @@ summary.lgb.Booster <- function(object, ...) {
#' , metric = "l2"
#' , metric = "l2"
#' , min_data = 1L
#' , min_data = 1L
#' , learning_rate = 1.0
#' , learning_rate = 1.0
#' , num_threads = 2L
#' )
#' )
#' valids <- list(test = dtest)
#' valids <- list(test = dtest)
#' model <- lgb.train(
#' model <- lgb.train(
...
@@ -1296,6 +1301,7 @@ lgb.load <- function(filename = NULL, model_str = NULL) {
...
@@ -1296,6 +1301,7 @@ lgb.load <- function(filename = NULL, model_str = NULL) {
#' , metric = "l2"
#' , metric = "l2"
#' , min_data = 1L
#' , min_data = 1L
#' , learning_rate = 1.0
#' , learning_rate = 1.0
#' , num_threads = 2L
#' )
#' )
#' valids <- list(test = dtest)
#' valids <- list(test = dtest)
#' model <- lgb.train(
#' model <- lgb.train(
...
@@ -1351,6 +1357,7 @@ lgb.save <- function(booster, filename, num_iteration = NULL) {
...
@@ -1351,6 +1357,7 @@ lgb.save <- function(booster, filename, num_iteration = NULL) {
#' , metric = "l2"
#' , metric = "l2"
#' , min_data = 1L
#' , min_data = 1L
#' , learning_rate = 1.0
#' , learning_rate = 1.0
#' , num_threads = 2L
#' )
#' )
#' valids <- list(test = dtest)
#' valids <- list(test = dtest)
#' model <- lgb.train(
#' model <- lgb.train(
...
@@ -1401,6 +1408,7 @@ lgb.dump <- function(booster, num_iteration = NULL) {
...
@@ -1401,6 +1408,7 @@ lgb.dump <- function(booster, num_iteration = NULL) {
#' , metric = "l2"
#' , metric = "l2"
#' , min_data = 1L
#' , min_data = 1L
#' , learning_rate = 1.0
#' , learning_rate = 1.0
#' , num_threads = 2L
#' )
#' )
#' valids <- list(test = dtest)
#' valids <- list(test = dtest)
#' model <- lgb.train(
#' model <- lgb.train(
...
...
R-package/R/lgb.cv.R
View file @
7dcbb8cd
...
@@ -59,6 +59,7 @@ CVBooster <- R6::R6Class(
...
@@ -59,6 +59,7 @@ CVBooster <- R6::R6Class(
#' , metric = "l2"
#' , metric = "l2"
#' , min_data = 1L
#' , min_data = 1L
#' , learning_rate = 1.0
#' , learning_rate = 1.0
#' , num_threads = 2L
#' )
#' )
#' model <- lgb.cv(
#' model <- lgb.cv(
#' params = params
#' params = params
...
...
R-package/R/lgb.importance.R
View file @
7dcbb8cd
...
@@ -24,6 +24,7 @@
...
@@ -24,6 +24,7 @@
#' , max_depth = -1L
#' , max_depth = -1L
#' , min_data_in_leaf = 1L
#' , min_data_in_leaf = 1L
#' , min_sum_hessian_in_leaf = 1.0
#' , min_sum_hessian_in_leaf = 1.0
#' , num_threads = 2L
#' )
#' )
#' model <- lgb.train(
#' model <- lgb.train(
#' params = params
#' params = params
...
...
R-package/R/lgb.interprete.R
View file @
7dcbb8cd
...
@@ -35,6 +35,7 @@
...
@@ -35,6 +35,7 @@
#' , max_depth = -1L
#' , max_depth = -1L
#' , min_data_in_leaf = 1L
#' , min_data_in_leaf = 1L
#' , min_sum_hessian_in_leaf = 1.0
#' , min_sum_hessian_in_leaf = 1.0
#' , num_threads = 2L
#' )
#' )
#' model <- lgb.train(
#' model <- lgb.train(
#' params = params
#' params = params
...
...
R-package/R/lgb.model.dt.tree.R
View file @
7dcbb8cd
...
@@ -40,6 +40,7 @@
...
@@ -40,6 +40,7 @@
#' , max_depth = -1L
#' , max_depth = -1L
#' , min_data_in_leaf = 1L
#' , min_data_in_leaf = 1L
#' , min_sum_hessian_in_leaf = 1.0
#' , min_sum_hessian_in_leaf = 1.0
#' , num_threads = 2L
#' )
#' )
#' model <- lgb.train(params, dtrain, 10L)
#' model <- lgb.train(params, dtrain, 10L)
#'
#'
...
...
R-package/R/lgb.plot.importance.R
View file @
7dcbb8cd
...
@@ -28,6 +28,7 @@
...
@@ -28,6 +28,7 @@
#' , learning_rate = 0.1
#' , learning_rate = 0.1
#' , min_data_in_leaf = 1L
#' , min_data_in_leaf = 1L
#' , min_sum_hessian_in_leaf = 1.0
#' , min_sum_hessian_in_leaf = 1.0
#' , num_threads = 2L
#' )
#' )
#'
#'
#' model <- lgb.train(
#' model <- lgb.train(
...
...
R-package/R/lgb.plot.interpretation.R
View file @
7dcbb8cd
...
@@ -39,6 +39,7 @@
...
@@ -39,6 +39,7 @@
#' , max_depth = -1L
#' , max_depth = -1L
#' , min_data_in_leaf = 1L
#' , min_data_in_leaf = 1L
#' , min_sum_hessian_in_leaf = 1.0
#' , min_sum_hessian_in_leaf = 1.0
#' , num_threads = 2L
#' )
#' )
#' model <- lgb.train(
#' model <- lgb.train(
#' params = params
#' params = params
...
...
R-package/R/lgb.restore_handle.R
View file @
7dcbb8cd
...
@@ -23,7 +23,9 @@
...
@@ -23,7 +23,9 @@
#' , agaricus.train$label
#' , agaricus.train$label
#' , params = list(objective = "binary")
#' , params = list(objective = "binary")
#' , nrounds = 5L
#' , nrounds = 5L
#' , verbose = 0)
#' , verbose = 0
#' , num_threads = 2L
#' )
#' fname <- tempfile(fileext="rds")
#' fname <- tempfile(fileext="rds")
#' saveRDS(model, fname)
#' saveRDS(model, fname)
#'
#'
...
...
R-package/R/lgb.train.R
View file @
7dcbb8cd
...
@@ -30,6 +30,7 @@
...
@@ -30,6 +30,7 @@
#' , metric = "l2"
#' , metric = "l2"
#' , min_data = 1L
#' , min_data = 1L
#' , learning_rate = 1.0
#' , learning_rate = 1.0
#' , num_threads = 2L
#' )
#' )
#' valids <- list(test = dtest)
#' valids <- list(test = dtest)
#' model <- lgb.train(
#' model <- lgb.train(
...
...
R-package/R/readRDS.lgb.Booster.R
View file @
7dcbb8cd
...
@@ -23,6 +23,7 @@
...
@@ -23,6 +23,7 @@
#' , metric = "l2"
#' , metric = "l2"
#' , min_data = 1L
#' , min_data = 1L
#' , learning_rate = 1.0
#' , learning_rate = 1.0
#' , num_threads = 2L
#' )
#' )
#' valids <- list(test = dtest)
#' valids <- list(test = dtest)
#' model <- lgb.train(
#' model <- lgb.train(
...
...
R-package/R/saveRDS.lgb.Booster.R
View file @
7dcbb8cd
...
@@ -33,6 +33,7 @@
...
@@ -33,6 +33,7 @@
#' , metric = "l2"
#' , metric = "l2"
#' , min_data = 1L
#' , min_data = 1L
#' , learning_rate = 1.0
#' , learning_rate = 1.0
#' , num_threads = 2L
#' )
#' )
#' valids <- list(test = dtest)
#' valids <- list(test = dtest)
#' model <- lgb.train(
#' model <- lgb.train(
...
...
R-package/man/lgb.configure_fast_predict.Rd
View file @
7dcbb8cd
...
@@ -119,7 +119,10 @@ data(mtcars)
...
@@ -119,7 +119,10 @@ data(mtcars)
X <- as.matrix(mtcars[, -1L])
X <- as.matrix(mtcars[, -1L])
y <- mtcars[, 1L]
y <- mtcars[, 1L]
dtrain <- lgb.Dataset(X, label = y, params = list(max_bin = 5L))
dtrain <- lgb.Dataset(X, label = y, params = list(max_bin = 5L))
params <- list(min_data_in_leaf = 2L)
params <- list(
min_data_in_leaf = 2L
, num_threads = 2L
)
model <- lgb.train(
model <- lgb.train(
params = params
params = params
, data = dtrain
, data = dtrain
...
...
R-package/man/lgb.cv.Rd
View file @
7dcbb8cd
...
@@ -160,6 +160,7 @@ params <- list(
...
@@ -160,6 +160,7 @@ params <- list(
,
metric
=
"l2"
,
metric
=
"l2"
,
min_data
=
1L
,
min_data
=
1L
,
learning_rate
=
1.0
,
learning_rate
=
1.0
,
num_threads
=
2L
)
)
model
<-
lgb
.
cv
(
model
<-
lgb
.
cv
(
params
=
params
params
=
params
...
...
R-package/man/lgb.dump.Rd
View file @
7dcbb8cd
...
@@ -31,6 +31,7 @@ params <- list(
...
@@ -31,6 +31,7 @@ params <- list(
,
metric
=
"l2"
,
metric
=
"l2"
,
min_data
=
1L
,
min_data
=
1L
,
learning_rate
=
1.0
,
learning_rate
=
1.0
,
num_threads
=
2L
)
)
valids
<-
list
(
test
=
dtest
)
valids
<-
list
(
test
=
dtest
)
model
<-
lgb
.
train
(
model
<-
lgb
.
train
(
...
...
R-package/man/lgb.get.eval.result.Rd
View file @
7dcbb8cd
...
@@ -45,6 +45,7 @@ params <- list(
...
@@ -45,6 +45,7 @@ params <- list(
,
metric
=
"l2"
,
metric
=
"l2"
,
min_data
=
1L
,
min_data
=
1L
,
learning_rate
=
1.0
,
learning_rate
=
1.0
,
num_threads
=
2L
)
)
valids
<-
list
(
test
=
dtest
)
valids
<-
list
(
test
=
dtest
)
model
<-
lgb
.
train
(
model
<-
lgb
.
train
(
...
...
R-package/man/lgb.importance.Rd
View file @
7dcbb8cd
...
@@ -35,6 +35,7 @@ params <- list(
...
@@ -35,6 +35,7 @@ params <- list(
, max_depth = -1L
, max_depth = -1L
, min_data_in_leaf = 1L
, min_data_in_leaf = 1L
, min_sum_hessian_in_leaf = 1.0
, min_sum_hessian_in_leaf = 1.0
, num_threads = 2L
)
)
model <- lgb.train(
model <- lgb.train(
params = params
params = params
...
...
R-package/man/lgb.interprete.Rd
View file @
7dcbb8cd
...
@@ -48,6 +48,7 @@ params <- list(
...
@@ -48,6 +48,7 @@ params <- list(
, max_depth = -1L
, max_depth = -1L
, min_data_in_leaf = 1L
, min_data_in_leaf = 1L
, min_sum_hessian_in_leaf = 1.0
, min_sum_hessian_in_leaf = 1.0
, num_threads = 2L
)
)
model <- lgb.train(
model <- lgb.train(
params = params
params = params
...
...
R-package/man/lgb.load.Rd
View file @
7dcbb8cd
...
@@ -31,6 +31,7 @@ params <- list(
...
@@ -31,6 +31,7 @@ params <- list(
,
metric
=
"l2"
,
metric
=
"l2"
,
min_data
=
1L
,
min_data
=
1L
,
learning_rate
=
1.0
,
learning_rate
=
1.0
,
num_threads
=
2L
)
)
valids
<-
list
(
test
=
dtest
)
valids
<-
list
(
test
=
dtest
)
model
<-
lgb
.
train
(
model
<-
lgb
.
train
(
...
...
R-package/man/lgb.model.dt.tree.Rd
View file @
7dcbb8cd
...
@@ -51,6 +51,7 @@ params <- list(
...
@@ -51,6 +51,7 @@ params <- list(
,
max_depth
=
-
1L
,
max_depth
=
-
1L
,
min_data_in_leaf
=
1L
,
min_data_in_leaf
=
1L
,
min_sum_hessian_in_leaf
=
1.0
,
min_sum_hessian_in_leaf
=
1.0
,
num_threads
=
2L
)
)
model
<-
lgb
.
train
(
params
,
dtrain
,
10L
)
model
<-
lgb
.
train
(
params
,
dtrain
,
10L
)
...
...
Prev
1
2
Next
Write
Preview
Markdown
is supported
0%
Try again
or
attach a new file
.
Attach a file
Cancel
You are about to add
0
people
to the discussion. Proceed with caution.
Finish editing this message first!
Cancel
Please
register
or
sign in
to comment