Commit 59c116fe authored by Guolin Ke's avatar Guolin Ke
Browse files

fix #276

parent eef4d2d0
...@@ -122,7 +122,7 @@ lgb.cv <- function(params=list(), data, nrounds = 10, ...@@ -122,7 +122,7 @@ lgb.cv <- function(params=list(), data, nrounds = 10,
data$update_params(params) data$update_params(params)
data$.__enclos_env__$private$set_predictor(predictor) data$.__enclos_env__$private$set_predictor(predictor)
if (!is.null(colnames)) { data$set_colnames(colnames) } if (!is.null(colnames)) { data$set_colnames(colnames) }
data$set_categorical_feature(categorical_feature) if (!is.null(categorical_feature)) { data$set_categorical_feature(categorical_feature) }
data$construct() data$construct()
if (!is.null(folds)) { if (!is.null(folds)) {
......
...@@ -100,7 +100,7 @@ lgb.train <- function(params = list(), data, nrounds = 10, ...@@ -100,7 +100,7 @@ lgb.train <- function(params = list(), data, nrounds = 10,
data$update_params(params) data$update_params(params)
data$.__enclos_env__$private$set_predictor(predictor) data$.__enclos_env__$private$set_predictor(predictor)
if (!is.null(colnames)) { data$set_colnames(colnames) } if (!is.null(colnames)) { data$set_colnames(colnames) }
data$set_categorical_feature(categorical_feature) if (!is.null(categorical_feature)) { data$set_categorical_feature(categorical_feature) }
data$construct() data$construct()
vaild_contain_train <- FALSE vaild_contain_train <- FALSE
train_data_name <- "train" train_data_name <- "train"
......
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