"vscode:/vscode.git/clone" did not exist on "646dfed9138480581b8a8b048c2865409b8838c8"
Commit 5256cda6 authored by Laurae's avatar Laurae Committed by Guolin Ke
Browse files

Fix #1988 missing file issue (#2000)

parent 462612b4
......@@ -69,7 +69,8 @@ my_data_test <- as.matrix(bank_test[, 1:16, with = FALSE])
# Creating the LightGBM dataset with categorical features
# The categorical features can be passed to lgb.train to not copy and paste a lot
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,
data = my_data_test,
label = bank_test$y)
......@@ -81,8 +82,7 @@ model <- lgb.train(list(objective = "binary",
learning_rate = 0.1,
min_data = 0,
min_hessian = 1,
max_depth = 2,
categorical_feature = c(2, 3, 4, 5, 7, 8, 9, 11, 16)),
max_depth = 2),
dtrain,
100,
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