search

PicNet / XGBoost.Net

.Net wrappers for the awesome XGBoost library
51 stars 18 forks source link

Add paramters for DART booster #23

Closed mdabros closed 6 years ago

mdabros commented 6 years ago

Hi @gatapia ,

I have added the missing parameters for when using the DART booster type. For now, I have simply hardcoded the default values in the XGBoostClassifier and XGBoostRegressor, so behavior should be the same.

I also added the updater parameter for the linear booster type, but for some reason it fails when running on GPU, so I ended up removing it again.

Best regards Mads