Alibi is an open source Python library aimed at machine learning model inspection and interpretation.
The focus of the library is to provide high-quality implementations of black-box, white-box, local and global
explanation methods for classification and regression models.
To take advantage of distributed computation of explanations, install alibi with ray:
pip install alibi[ray]
For SHAP support, install alibi as follows:
pip install alibi[shap]
With conda
To install from conda-forge it is recommended to use mamba,
which can be installed to the base conda enviroment with:
conda install mamba -n base -c conda-forge
For the standard Alibi install:
mamba install -c conda-forge alibi
For distributed computing support:
mamba install -c conda-forge alibi ray
For SHAP support:
mamba install -c conda-forge alibi shap
Usage
The alibi explanation API takes inspiration from scikit-learn, consisting of distinct initialize,
fit and explain steps. We will use the AnchorTabular
explainer to illustrate the API:
fromalibi.explainersimportAnchorTabular# initialize and fit explainer by passing a prediction function and any other required argumentsexplainer=AnchorTabular(predict_fn, feature_names=feature_names, category_map=category_map)
explainer.fit(X_train)
# explain an instanceexplanation=explainer.explain(x)
The explanation returned is an Explanation object with attributes meta and data. meta is a dictionary
containing the explainer metadata and any hyperparameters and data is a dictionary containing everything
related to the computed explanation. For example, for the Anchor algorithm the explanation can be accessed
via explanation.data['anchor'] (or explanation.anchor). The exact details of available fields varies
from method to method so we encourage the reader to become familiar with the
types of methods supported.
Supported Methods
The following tables summarize the possible use cases for each method.
If you use alibi in your research, please consider citing it.
BibTeX entry:
@article{JMLR:v22:21-0017,
author = {Janis Klaise and Arnaud Van Looveren and Giovanni Vacanti and Alexandru Coca},
title = {Alibi Explain: Algorithms for Explaining Machine Learning Models},
journal = {Journal of Machine Learning Research},
year = {2021},
volume = {22},
number = {181},
pages = {1-7},
url = {http://jmlr.org/papers/v22/21-0017.html}
}
请发表评论