Add TreeSHAP dependencies to Alibi-Explain runtime image

SeldonIO / MLServer

An inference server for your machine learning models, including support for multiple frameworks, multi-model serving and more

https://mlserver.readthedocs.io/en/latest/

Apache License 2.0

719 stars 183 forks source link

Add TreeSHAP dependencies to Alibi-Explain runtime image #796

Open adriangonz opened 2 years ago

adriangonz commented 2 years ago

The TreeSHAP explainer needs access to the underlying model instance. Therefore, to ensure TreeSHAP works out-of-the-box, we'll need to ensure the following libraries are present on the Alibi Explain runtime image:

XGBoost
LightGBM
CatBoost
SKLearn

ascillitoe commented 1 year ago

https://github.com/SeldonIO/MLServer/pull/1279 adds XGBoost, LightGBM and SKLearn to the runtime deps. CatBoost is still TODO, since for consistency we might want to wait for an mlserver_catboost runtime to be added first.