Skip to content

GitLab

Commit 5256cda6 authored by Laurae's avatar Laurae Committed by Guolin Ke
Browse files

Fix #1988 missing file issue (#2000)

parent 462612b4
Hide whitespace changes
Inline Side-by-side
Showing with 3 additions and 3 deletions
R-package/demo/categorical_features_rules.R
View file @ 5256cda6
...@@ -69,7 +69,8 @@ my_data_test <- as.matrix(bank_test[, 1:16, with = FALSE]) ...@@ -69,7 +69,8 @@ my_data_test <- as.matrix(bank_test[, 1:16, with = FALSE])
# Creating the LightGBM dataset with categorical features # Creating the LightGBM dataset with categorical features
# The categorical features can be passed to lgb.train to not copy and paste a lot # The categorical features can be passed to lgb.train to not copy and paste a lot
dtrain <- lgb.Dataset(data = my_data_train, dtrain <- lgb.Dataset(data = my_data_train,
label = bank_train$y) label = bank_train$y,
categorical_feature = c(2, 3, 4, 5, 7, 8, 9, 11, 16))
dtest <- lgb.Dataset.create.valid(dtrain, dtest <- lgb.Dataset.create.valid(dtrain,
data = my_data_test, data = my_data_test,
label = bank_test$y) label = bank_test$y)
...@@ -81,8 +82,7 @@ model <- lgb.train(list(objective = "binary", ...@@ -81,8 +82,7 @@ model <- lgb.train(list(objective = "binary",
learning_rate = 0.1, learning_rate = 0.1,
min_data = 0, min_data = 0,
min_hessian = 1, min_hessian = 1,
max_depth = 2, max_depth = 2),
categorical_feature = c(2, 3, 4, 5, 7, 8, 9, 11, 16)),
dtrain, dtrain,
100, 100,
valids = list(train = dtrain, valid = dtest)) valids = list(train = dtrain, valid = dtest))
......
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