gsganden / model_inspector

A uniform interface to a curated set of methods for inspecting machine learning models
https://gsganden.github.io/model_inspector/
Apache License 2.0
4 stars 0 forks source link

Support `xgboost` #30

Closed gsganden closed 1 year ago

gsganden commented 1 year ago

We currently support scikit-learn models only. Adding support for xgboost might not be too difficult and would make the library more relevant for practitioners.

gsganden commented 1 year ago

I think we can use Model Inspector with XGBoost by way of the xgboost library's sklearn wrapper, but I haven't confirmed that, we might not want to use the wrapper, and we might want to add some XGBoost-specific methods (e.g. learning curves).