-
-
Notifications
You must be signed in to change notification settings - Fork 407
Update gbm learners #2651
New issue
Have a question about this project? Sign up for a free GitHub account to open an issue and contact its maintainers and the community.
By clicking “Sign up for GitHub”, you agree to our terms of service and privacy statement. We’ll occasionally send you account related emails.
Already on GitHub? Sign in to your account
Update gbm learners #2651
Conversation
R/RLearner_classif_gbm.R
Outdated
| note = "`keep.data` is set to FALSE to reduce memory requirements. Note on param 'distribution': gbm will select 'bernoulli' by default for 2 classes, and 'multinomial' for multiclass problems. The latter is the only setting that works for > 2 classes. It is recommended to NOT use the 'multinomial' setting, see https://github.com/mlr-org/mlr/issues/2612", | ||
| note = paste0(collapse = "", c("`keep.data` is set to FALSE to reduce memory requirements.\n", | ||
| "Note on param 'distribution': gbm will select 'bernoulli' by default for 2 classes, and 'multinomial' for multiclass problems.\n", | ||
| "The latter is the only setting that works for > 2 classes. It is recommended to NOT use the 'multinomial' setting, see https://github.com/mlr-org/mlr/issues/2612.", |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
This is confusing. There's no "multinomial" value anymore, and the default doesn't depend on the task here.
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Yes, good catch. Removed the note about "multinomial" entirely as it does not apply anymore.
R/RLearner_regr_gbm.R
Outdated
| name = "Gradient Boosting Machine", | ||
| short.name = "gbm", | ||
| note = '`keep.data` is set to FALSE to reduce memory requirements, `distribution` has been set to `"gaussian"` by default.', | ||
| note = poaste0(collapse = "", c('`keep.data` is set to FALSE to reduce memory requirements, `distribution` has been set to `"gaussian"` by default.', |
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Mentioned default value for distribution is incorrect. Same note on distribution as for classification learner here?
There was a problem hiding this comment.
Choose a reason for hiding this comment
The reason will be displayed to describe this comment to others. Learn more.
Ah I was lazy here. Sorry. Default now "gaussian" and the classif learner was changed as well.
* upd gbm distribution argument * upd shrinkage and distribtion argument * update learner note * document n.cores change * add NEWS * correct regr.gbm distribution param * clear note about 'multinomial' param for classif.gbm * fix classif_gbm test * add multionomial option for distribution param again * Deploy from Travis build 14223 [ci skip] Build URL: https://travis-ci.org/mlr-org/mlr/builds/597271702 Commit: 2e4bceb
closes #2642
shrinkagefrom 0.001 to 0.1n.coresand suppress parallelization by defaultdistributionwith the actual package defaultsdistribution