MCPcopy Index your code
hub / github.com/SeldonIO/alibi

github.com/SeldonIO/alibi @v0.9.6

Chat with this repo
repository ↗ · DeepWiki ↗ · release v0.9.6 ↗ · + Follow
1,397 symbols 5,225 edges 142 files 939 documented · 67%
What it actually does AI analysis from the code graph — generated when you open this
loading…
README

Alibi Logo

Build Status Documentation Status codecov PyPI - Python Version PyPI - Package Version Conda (channel only) GitHub - License Slack channel


Alibi is a 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. * Documentation

If you're interested in outlier detection, concept drift or adversarial instance detection, check out our sister project alibi-detect.

Anchor explanations for images Integrated Gradients for text
Counterfactual examples Accumulated Local Effects

Table of Contents

Installation and Usage

Alibi can be installed from:

  • PyPI or GitHub source (with pip)
  • Anaconda (with conda/mamba)

With pip

  • Alibi can be installed from PyPI:

bash pip install alibi

  • Alternatively, the development version can be installed: bash pip install git+https://github.com/SeldonIO/alibi.git

  • To take advantage of distributed computation of explanations, install alibi with ray: bash pip install alibi[ray]

  • For SHAP support, install alibi as follows: bash 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: bash mamba install -c conda-forge alibi

  • For distributed computing support: bash mamba install -c conda-forge alibi ray

  • For SHAP support: bash 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:

from alibi.explainers import AnchorTabular

# initialize and fit explainer by passing a prediction function and any other required arguments
explainer = AnchorTabular(predict_fn, feature_names=feature_names, category_map=category_map)
explainer.fit(X_train)

# explain an instance
explanation = 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.

Model Explanations

Method Models Explanations Classification Regression Tabular Text Images Categorical features Train set required Distributed
ALE BB global
Partial Dependence BB WB global
PD Variance BB WB global
Permutation Importance BB global
Anchors BB local For Tabular
CEM BB* TF/Keras local Optional
Counterfactuals BB* TF/Keras local No
Prototype Counterfactuals BB* TF/Keras local Optional
Counterfactuals with RL BB local
Integrated Gradients TF/Keras local Optional
Kernel SHAP BB local

global | ✔ | ✔ | ✔ | | | ✔ | ✔ | ✔ | | Tree SHAP | WB | local

global | ✔ | ✔ | ✔ | | | ✔ | Optional | | | Similarity explanations | WB | local | ✔ | ✔ | ✔ | ✔ | ✔ | ✔ | ✔ | |

Model Confidence

These algorithms provide instance-specific scores measuring the model confidence for making a particular prediction.

Method Models Classification Regression Tabular Text Images Categorical Features Train set required
Trust Scores BB ✔(1) ✔(2) Yes
Linearity Measure BB Optional

Key: - BB - black-box (only require a prediction function) - BB* - black-box but assume model is differentiable - WB - requires white-box model access. There may be limitations on models supported - TF/Keras - TensorFlow models via the Keras API - Local - instance specific explanation, why was this prediction made? - Global - explains the model with respect to a set of instances - (1) - depending on model - (2) - may require dimensionality reduction

Prototypes

These algorithms provide a distilled view of the dataset and help construct a 1-KNN interpretable classifier.

Method Classification Regression Tabular Text Images Categorical Features Train set labels
ProtoSelect Optional

References and Examples

Core symbols most depended-on inside this repo

append
called by 183
alibi/explainers/cfrl_base.py
argmax
called by 78
alibi/explainers/similarity/backends/pytorch/base.py
predict
called by 48
alibi/tests/utils.py
fit
called by 43
alibi/explainers/similarity/grad.py
seed
called by 35
alibi/explainers/anchors/language_model_text_sampler.py
explain
called by 23
alibi/explainers/anchors/anchor_text.py
import_optional
called by 22
alibi/utils/missing_optional_dependency.py
to_numpy
called by 21
alibi/explainers/similarity/backends/pytorch/base.py

Shape

Function 675
Method 510
Class 129
Route 83

Languages

Python100%

Modules by API surface

alibi/models/pytorch/tests/test_model.py49 symbols
alibi/explainers/tests/test_partial_dependence.py47 symbols
alibi/explainers/tests/test_integrated_gradients.py43 symbols
alibi/explainers/tests/test_shap_wrappers.py42 symbols
alibi/explainers/tests/test_pd_variance.py38 symbols
alibi/explainers/shap_wrappers.py35 symbols
alibi/tests/test_saving.py34 symbols
alibi/utils/distributed.py32 symbols
alibi/explainers/partial_dependence.py31 symbols
alibi/explainers/integrated_gradients.py29 symbols
alibi/explainers/cfrl_base.py29 symbols
alibi/explainers/anchors/anchor_tabular.py27 symbols

For agents

$ claude mcp add alibi \
  -- python -m otcore.mcp_server <graph>

⬇ download graph artifact