Ecosyste.ms: Awesome

An open API service indexing awesome lists of open source software.

Awesome Lists | Featured Topics | Projects

https://github.com/elo7/autoshap

AutoSHAP is a framework to automate explainable machine learning
https://github.com/elo7/autoshap

Last synced: about 2 months ago
JSON representation

AutoSHAP is a framework to automate explainable machine learning

Awesome Lists containing this project

README

        

# AutoSHAP

AutoSHAP is a library still in development by the [Elo7](https://www.elo7.com.br) data science and search teams.

AutoSHAP is a simple framework built on top of [SHAP](https://github.com/slundberg/shap) to make it simple to engineers access explanations in a daily basis setting just a few parameters. It has two main classes, _FullPipeline_ and _SHAPViews_. The first runs the entire pipeline depicted in image bellow, i.e., it instances a [XGBoost](https://xgboost.readthedocs.io/en/latest/) model, makes hyperparameter optimization, computes the Shapley values with the best model, saves the XGBoost model and data and make plots as a result.

images/autoshap-pipeline

The second class, _SHAPViews_, allows one to see interactive plots of features and Shapley values. This is specially useful to analyze the distribution of some feature values while filtering them by some chosen Shapley Values.

## Installation

Clone this repository,
```bash
git clone [email protected]:elo7/AutoSHAP.git
```
and inside it use pip to install:
```bash
pip install .
```

## Examples

You can check application examples in the [notebooks](notebooks/)