explainerdashboard  by oegedijk

Explainable AI dashboard for ML model insights

created 5 years ago
2,415 stars

Top 19.5% on sourcepulse

GitHubView on GitHub
1 Expert Loves This Project
Project Summary

This library provides a framework for building interactive dashboards to explain the inner workings of machine learning models, targeting data scientists and non-technical stakeholders. It simplifies the process of visualizing model performance, feature importances, and individual predictions, making complex AI models more transparent and interpretable.

How It Works

The library leverages Plotly Dash to create interactive web applications. It integrates with popular ML libraries (scikit-learn, XGBoost, LightGBM, CatBoost, Skorch) and calculates explainability metrics like SHAP values, permutation importances, and partial dependence plots. Its modular design allows for custom dashboard layouts and the aggregation of multiple dashboards into an "ExplainerHub."

Quick Start & Requirements

  • Install via pip: pip install explainerdashboard or conda: conda install -c conda-forge explainerdashboard.
  • Supports scikit-learn, xgboost, catboost, lightgbm, and skorch models.
  • Official documentation: explainerdashboard.readthedocs.io
  • Example notebook: launching dashboards

Highlighted Details

  • Interactive exploration of model performance, feature importances, SHAP values, and "what-if" scenarios.
  • Supports visualization of individual decision trees for tree-based models.
  • Can export dashboards to static HTML for easy sharing and CI/CD integration.
  • Modular design allows for custom layouts and integration into larger ExplainerHubs.

Maintenance & Community

Licensing & Compatibility

  • License: Not explicitly stated in the README.

Limitations & Caveats

Calculations for SHAP interaction values and permutation importances can be slow for large datasets; options exist to disable or approximate these. Memory usage can be substantial for large datasets, with suggestions for optimization like using float32 precision or external data storage.

Health Check
Last commit

1 day ago

Responsiveness

1 week

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

Explore Similar Projects

Feedback? Help us improve.