Skip to main content

XAISuite: Training and Explanation Generation Utilities for Machine Learning Models

Reason this release was yanked:

This is an unstable version without complete functionalities.

Project description



XAISuite: Training and Explaining Machine Learning Models

Table of Contents

  1. Introduction
  2. Installation
  3. Getting Started
  4. How to Contribute
  5. Technical Report and Citing XAISuite

Introduction

XAISuite (Explanatory Artificial Intelligence Suite) is a library for training and explaining machine learning models for tabular datasets in Python. It provides a unified interface for training any sklearn model using just a line of code and allows users to easily compare the results of different explainers!

XAISuite accomplishes machine learning model training and explanation generation in three steps: (1) data loading, (2) model training and (3) explanation generation. Each of these steps are delved into more detail in our documentation and in the demo tutorials.

Basic Flowchart of how XAISuite works

XAISuite was created as a helper library to [this paper](insert link), which studied the difference in SHAP and LIME explanations for different models on tabular datasets.

Installation

You can install the XAI Suite through PyPI:

pip install XAISuite

Getting Started

For example code and an introduction to the library, see the Demo Folder.

If you are looking for a model or dataset to use, sklearn has several cool options.

How to Contribute

We welcome the contribution from the open-source community to improve the library!

To add a new functionality into the library or point out a flaw, please create a new issue on Github. We'll try to look into your requests as soon as we can.

Technical Report and Citing XAISuite

A paper proposing and using XAISuite to compare explanatory methods is still in pre-publication. Use the following BibTex to cite XAISuite:

@article{mitra2022-xaisuite,
  author    = {Shreyan Mitra and Leilani Gilpin},
  title     = {Comparison of SHAP and LIME Explanations for Supervised
Machine Learning Models Trained on Tabular Datasets},
  year      = {2022},
  doi       = {},
  url       = {},
  archivePrefix = {},
  eprint    = {},
}

Contact Us

If you have any questions, comments or suggestions, please do not hesitate to contact us at xaisuite@gmail.com

License

This work is licensed under a BSD 3-Clause License.

Project details


Release history Release notifications | RSS feed

Download files

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

Source Distribution

XAISuite-0.8.3.3.tar.gz (11.3 kB view hashes)

Uploaded Source

Built Distribution

XAISuite-0.8.3.3-py3-none-any.whl (12.8 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