search

erdogant / hgboost

hgboost is a python package for hyper-parameter optimization for xgboost, catboost or lightboost using cross-validation, and evaluating the results on an independent validation set. hgboost can be applied for classification and regression tasks.
http://erdogant.github.io/hgboost
Other
57 stars 17 forks source link

Hgboost on GPU #11

Closed ChandanVerma closed 2 years ago

ChandanVerma commented 2 years ago

Does this package has the capability to operate on GPU? if yes, can you specify where to specify the parameter?

erdogant commented 2 years ago

I added GPU support but this is not always possible/allowed. Check out the documentation and give it a try with the latest version.

pip install -U hgboost