eli5  by eli5-org

Debugging library for inspecting ML classifiers and explaining predictions

created 4 years ago
301 stars

Top 89.6% on sourcepulse

GitHubView on GitHub
1 Expert Loves This Project
Project Summary

ELI5 is a Python library designed to help debug machine learning classifiers and explain their predictions. It targets data scientists and ML engineers working with various frameworks, offering insights into model behavior and facilitating easier debugging. The primary benefit is enhanced model interpretability and faster identification of issues.

How It Works

ELI5 provides a unified interface for explaining model internals and predictions across multiple ML libraries. It supports direct inspection of weights and feature importances for linear models, decision trees, and tree ensembles from scikit-learn, XGBoost, LightGBM, CatBoost, and lightning. For black-box models, it integrates LIME for text explanations and offers permutation importance. Explanations can be rendered as text, HTML, pandas DataFrames, or JSON for flexible integration.

Quick Start & Requirements

  • Install via pip: pip install eli5
  • Requires Python. Supports various ML frameworks including scikit-learn, Keras, XGBoost, LightGBM, CatBoost, lightning, sklearn-crfsuite, and OpenAI Python client.
  • Documentation available at: https://eli5.readthedocs.io/

Highlighted Details

  • Supports explanation of scikit-learn pipelines and FeatureUnions.
  • Can debug HashingVectorizer issues by undoing hashing.
  • Offers Grad-CAM visualizations for Keras image classifiers.
  • Provides token probability explanations for OpenAI LLM predictions.

Maintenance & Community

Licensing & Compatibility

  • License: MIT.
  • Compatible with commercial and closed-source projects.

Limitations & Caveats

The library's support for non-text data and arbitrary black-box classifiers using LIME is noted as experimental.

Health Check
Last commit

3 months ago

Responsiveness

Inactive

Pull Requests (30d)
0
Issues (30d)
0
Star History
14 stars in the last 90 days

Explore Similar Projects

Starred by Dominik Moritz Dominik Moritz(Professor at CMU; ML Researcher at Apple), Stas Bekman Stas Bekman(Author of Machine Learning Engineering Open Book; Research Engineer at Snowflake), and
2 more.

ecco by jalammar

0%
2k
Python library for interactive NLP model visualization in Jupyter notebooks
created 4 years ago
updated 11 months ago
Starred by Chip Huyen Chip Huyen(Author of AI Engineering, Designing Machine Learning Systems), Elie Bursztein Elie Bursztein(Cybersecurity Lead at Google DeepMind), and
1 more.

alibi by SeldonIO

0.1%
3k
Python library for ML model inspection and interpretation
created 6 years ago
updated 1 month ago
Feedback? Help us improve.