eli5  by eli5-org

Debugging library for inspecting ML classifiers and explaining predictions

Created 4 years ago
307 stars

Top 87.3% 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

5 months ago

Responsiveness

Inactive

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

Explore Similar Projects

Starred by Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems"), Travis Addair Travis Addair(Cofounder of Predibase), and
4 more.

alibi by SeldonIO

0.1%
3k
Python library for ML model inspection and interpretation
Created 6 years ago
Updated 14 hours ago
Starred by Chip Huyen Chip Huyen(Author of "AI Engineering", "Designing Machine Learning Systems"), Gabriel Almeida Gabriel Almeida(Cofounder of Langflow), and
5 more.

lit by PAIR-code

0.1%
4k
Interactive ML model analysis tool for understanding model behavior
Created 5 years ago
Updated 3 weeks ago
Feedback? Help us improve.