search

IntelPython / scikit-learn_bench

scikit-learn_bench benchmarks various implementations of machine learning algorithms across data analytics frameworks. It currently support the scikit-learn, DAAL4PY, cuML, and XGBoost frameworks for commonly used machine learning algorithms.
Apache License 2.0
113 stars 69 forks source link

HistGradientBoostingEstimator #82

Closed amueller closed 4 months ago

amueller commented 3 years ago

Hi! Is there a reason HistGradientBoostingEstimator from sklearn is not included in the benchmark? It should be about as fast as XGBoost.

ShvetsKS commented 3 years ago

@amueller thanks a lot for reporting the issue! We are working on adding the HistGradientBoostingClassifier, HistGradientBoostingRegressor into current benchmarks.

amueller commented 3 years ago

Awesome, thanks!

napetrov commented 1 year ago

Will be addressed by https://github.com/IntelPython/scikit-learn_bench/pull/133