"git@developer.sourcefind.cn:tianlh/lightgbm-dcu.git" did not exist on "350b461220b2f129bd0993baed2c3989a793b4ed"
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,
data$update_params(params)
data$.__enclos_env__$private$set_predictor(predictor)
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()
if (!is.null(folds)) {
......
......@@ -100,7 +100,7 @@ lgb.train <- function(params = list(), data, nrounds = 10,
data$update_params(params)
data$.__enclos_env__$private$set_predictor(predictor)
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()
vaild_contain_train <- FALSE
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