Add necessary updates to run SHAP value benchmarks

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

111 stars 69 forks source link

Add necessary updates to run SHAP value benchmarks #142

Closed ahuber21 closed 11 months ago

ahuber21 commented 11 months ago

Support for SHAP value calculation measurement (XGBoost only)
Update XGBoost to user the newer d4p.mb.convert_model() API