Unverified Commit 0c1c36cd authored by Aditya Trivedi's avatar Aditya Trivedi Committed by GitHub
Browse files

[R-package] Updated lgb.train.R with keyword arguments (#3452)

* [R-package] Updated lgb.train.R with keyword arguments

* [R-package] Updated lgb.train.R with keyword arguments

* [R-package] Updated lgb.train.R with keyword arguments
parent e97424be
...@@ -84,8 +84,8 @@ lgb.train <- function(params = list(), ...@@ -84,8 +84,8 @@ lgb.train <- function(params = list(),
additional_params <- list(...) additional_params <- list(...)
params <- append(params, additional_params) params <- append(params, additional_params)
params$verbose <- verbose params$verbose <- verbose
params <- lgb.check.obj(params, obj) params <- lgb.check.obj(params = params, obj = obj)
params <- lgb.check.eval(params, eval) params <- lgb.check.eval(params = params, eval = eval)
fobj <- NULL fobj <- NULL
eval_functions <- list(NULL) eval_functions <- list(NULL)
...@@ -238,8 +238,8 @@ lgb.train <- function(params = list(), ...@@ -238,8 +238,8 @@ lgb.train <- function(params = list(),
# If user supplied early_stopping_rounds, add the early stopping callback # If user supplied early_stopping_rounds, add the early stopping callback
if (using_early_stopping) { if (using_early_stopping) {
callbacks <- add.cb( callbacks <- add.cb(
callbacks cb_list = callbacks
, cb.early.stop( , cb = cb.early.stop(
stopping_rounds = early_stopping_rounds stopping_rounds = early_stopping_rounds
, first_metric_only = isTRUE(params[["first_metric_only"]]) , first_metric_only = isTRUE(params[["first_metric_only"]])
, verbose = verbose , verbose = verbose
...@@ -247,16 +247,16 @@ lgb.train <- function(params = list(), ...@@ -247,16 +247,16 @@ lgb.train <- function(params = list(),
) )
} }
cb <- categorize.callbacks(callbacks) cb <- categorize.callbacks(cb_list = callbacks)
# Construct booster with datasets # Construct booster with datasets
booster <- Booster$new(params = params, train_set = data) booster <- Booster$new(params = params, train_set = data)
if (valid_contain_train) { if (valid_contain_train) {
booster$set_train_data_name(train_data_name) booster$set_train_data_name(name = train_data_name)
} }
for (key in names(reduced_valid_sets)) { for (key in names(reduced_valid_sets)) {
booster$add_valid(reduced_valid_sets[[key]], key) booster$add_valid(data = reduced_valid_sets[[key]], name = key)
} }
# Callback env # Callback env
......
Markdown is supported
0% or .
You are about to add 0 people to the discussion. Proceed with caution.
Finish editing this message first!
Please register or to comment