Skip to main content

A benchmark for feature attribution techniques

Project description

AttriBench is a Pytorch-based implementation of several metrics for the evaluation of feature attribution maps and methods. AttriBench provides a functional and an object-oriented API for the computation of these metrics, along with a set of utility functions for the necessary preparations (e.g. computing attribution maps) as well as for the visualization of the results.

The functional API is generally easier to use, and can be used to get started quickly if the scale of the evaluation is not too large. The object-oriented API is more flexible and can use multiple GPUs for evaluation of large datasets.

For more information, see the documentation.

Installation

AttriBench can be installed from PyPI using pip:

pip install attribench

Project details


Download files

Download the file for your platform. If you're not sure which to choose, learn more about installing packages.

Source Distribution

attribench-0.1.9.tar.gz (70.8 kB view hashes)

Uploaded Source

Built Distribution

attribench-0.1.9-py3-none-any.whl (122.4 kB view hashes)

Uploaded Python 3

Supported by

AWS AWS Cloud computing and Security Sponsor Datadog Datadog Monitoring Fastly Fastly CDN Google Google Download Analytics Microsoft Microsoft PSF Sponsor Pingdom Pingdom Monitoring Sentry Sentry Error logging StatusPage StatusPage Status page