mlr3learners / mlr3learners.lightgbm

Learners from {lightgbm} for mlr3
GNU Lesser General Public License v2.1
9 stars 3 forks source link

Issue in passing weight to lgb.Dataset #7

Closed kapsner closed 4 years ago

kapsner commented 4 years ago

Need to extract weight col from task$data() and pass it to lgb.Dataset

https://github.com/mlr3learners/mlr3learners.lightgbm/blob/development/R/LearnerClassifLightGBM.R#L668

kapsner commented 4 years ago

task$weights$weight already returns a vector with class weights - noting to be changed.