LGApr 3, 2024

Effector: A Python package for regional explanations

arXiv:2404.02629v24 citationsh-index: 48Has Code
AI Analysis

This tool helps data scientists and researchers improve model interpretability for tabular data, but it is incremental as it builds on existing methods like PDP and ALE.

Effector is a Python package that provides global and regional feature effect methods for interpreting tabular machine learning models, addressing misleading interpretations from feature interactions by partitioning input spaces and offering efficient implementations under a unified API.

Effector is a Python package for interpreting machine learning (ML) models that are trained on tabular data through global and regional feature effects. Global effects, like Partial Dependence Plot (PDP) and Accumulated Local Effects (ALE), are widely used for explaining tabular ML models due to their simplicity -- each feature's average influence on the prediction is summarized by a single 1D plot. However, when features are interacting, global effects can be misleading. Regional effects address this by partitioning the input space into disjoint subregions with minimal interactions within each and computing a separate regional effect per subspace. Regional effects are then visualized by a set of 1D plots per feature. Effector provides efficient implementations of state-of-the-art global and regional feature effects methods under a unified API. The package integrates seamlessly with major ML libraries like scikit-learn and PyTorch. It is designed to be modular and extensible, and comes with comprehensive documentation and tutorials. Effector is an open-source project publicly available on Github at https://github.com/givasile/effector.

Code Implementations1 repo
Foundations

The foundational work for this paper's niche, ranked by how specifically the neighbourhood builds on it — not by global fame.

Your Notes