Commit 427a0e20 authored by James Lamb's avatar James Lamb Committed by Nikita Titov
Browse files

[R-package] fixed typo for variable valid_contain_train (#2653)

parent 7b411bdd
...@@ -157,7 +157,7 @@ lgb.train <- function(params = list(), ...@@ -157,7 +157,7 @@ lgb.train <- function(params = list(),
# Construct datasets, if needed # Construct datasets, if needed
data$construct() data$construct()
vaild_contain_train <- FALSE valid_contain_train <- FALSE
train_data_name <- "train" train_data_name <- "train"
reduced_valid_sets <- list() reduced_valid_sets <- list()
...@@ -172,7 +172,7 @@ lgb.train <- function(params = list(), ...@@ -172,7 +172,7 @@ lgb.train <- function(params = list(),
# Check for duplicate train/validation dataset # Check for duplicate train/validation dataset
if (identical(data, valid_data)) { if (identical(data, valid_data)) {
vaild_contain_train <- TRUE valid_contain_train <- TRUE
train_data_name <- key train_data_name <- key
next next
} }
...@@ -249,7 +249,9 @@ lgb.train <- function(params = list(), ...@@ -249,7 +249,9 @@ lgb.train <- function(params = list(),
# 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 (vaild_contain_train) { booster$set_train_data_name(train_data_name) } if (valid_contain_train) {
booster$set_train_data_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(reduced_valid_sets[[key]], key)
} }
...@@ -282,7 +284,7 @@ lgb.train <- function(params = list(), ...@@ -282,7 +284,7 @@ lgb.train <- function(params = list(),
if (length(valids) > 0L) { if (length(valids) > 0L) {
# Validation has training dataset? # Validation has training dataset?
if (vaild_contain_train) { if (valid_contain_train) {
eval_list <- append(eval_list, booster$eval_train(feval = feval)) eval_list <- append(eval_list, booster$eval_train(feval = feval))
} }
......
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