Does thundergbm support early stopping?

Xtra-Computing / thundergbm

ThunderGBM: Fast GBDTs and Random Forests on GPUs

Apache License 2.0

689 stars 85 forks source link

Does thundergbm support early stopping? #9

Open rookiebird opened 5 years ago

rookiebird commented 5 years ago

Hi,in your document ,it says that the parameters is identical to XBoost. So Does thunderGBM support early stopping? It seems that 'thundergbm.sparse_train_scikit' doesn't support this parameter.

zeyiwen commented 5 years ago

Early stopping hasn't been supported yet. It would be awesome if you or others in the community can help regarding this issue.

We are trying our best to be compatible with XGBoost, so for all the existing parameters in ThunderGBM, we use the identical names to XGBoost.