Displaying 1 result from an estimated 1 matches for "min_child_weight".
2017 Nov 24
0
Using bartMachine with the caret package
...using caret to create xgbTree
models. The caret train() function has a tuneGrid parameter which
takes a list set up like so:
tune.grid <- expand.grid(eta = c(0.05, 0.075, 0.1),
nrounds = c(50, 75, 100),
max_depth = 6:8,
min_child_weight = c(2, 2.25, 2.5),
colsample_bytree = (3:5)/10,
gamma = 0, subsample = 1)
That approach also worked with my data. By making the corresponding
adjustments, I was also successful with gbm, bstTree and extraTree
models but I can't get it to work...