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

Model converters benchmark update #86

Open RukhovichIV opened 3 years ago

RukhovichIV commented 3 years ago

Here's some important updates for xgb and lgbm model converters bench + better parameters for gpu measurements to avoid Out Of Memory error on v100 machines