Skip to main content

SDK API to explain models, generate counterfactual examples, analyze causal effects and analyze errors in Machine Learning models.

Project description

Trustworthy AI Model Analysis SDK for Python

This package has been tested with Python 3.7, 3.8, 3.9 and 3.10

The Trustworthy AI Model Analysis SDK enables users to analyze their machine learning models in one API. Users will be able to analyze errors, explain the most important features, compute counterfactuals and run causal analysis using a single API.

Highlights of the package include:

  • explainer.add() explains the model
  • counterfactuals.add() computes counterfactuals
  • error_analysis.add() runs error analysis
  • causal.add() runs causal analysis

Supported scenarios, models and datasets

trustworthyai supports computation of Trustworthy AI insights for scikit-learn models that are ttained on pandas.DataFrame. The trustworthyai accept both models and pipelines as input as long as the model or pipeline implements a predict or predict_proba function that conforms to the scikit-learn convention. If not compatible, you can wrap your model's prediction function into a wrapper class that transforms the output into the format that is supported (predict or predict_proba of scikit-learn), and pass that wrapper class to modules in trustworthyai.

Currently, we support datasets having numerical and categorical features. The following table provides the scenarios supported for each of the four trustworthy AI insights:-

TAI insight Binary classification Multi-class classification Multilabel classification Regression Timeseries forecasting Categorical features Text features Image Features Recommender Systems Reinforcement Learning
Explainability Yes Yes No Yes No Yes No No No No
Error Analysis Yes Yes No Yes No Yes No No No No
Causal Analysis Yes No No Yes No Yes (max 5 features due to expensiveness) No No No No
Counterfactual Yes Yes No Yes No Yes No No No No

The source code can be found here: https://github.com/affectlog/trustworthy-ai-toolbox/tree/main/trustworthyai

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

trustworthyai-0.0.1.tar.gz (112.8 kB view details)

Uploaded Source

Built Distribution

If you're not sure about the file name format, learn more about wheel file names.

trustworthyai-0.0.1-py3-none-any.whl (153.7 kB view details)

Uploaded Python 3

File details

Details for the file trustworthyai-0.0.1.tar.gz.

File metadata

  • Download URL: trustworthyai-0.0.1.tar.gz
  • Upload date:
  • Size: 112.8 kB
  • Tags: Source
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.11.10

File hashes

Hashes for trustworthyai-0.0.1.tar.gz
Algorithm Hash digest
SHA256 1f8842232398c8fe670d907c3868ccbdf1ede0733aea686123c52937b8e2756e
MD5 b44f578e26b3a55522581420341ba51d
BLAKE2b-256 7a99cee06d6612fa363fe3a31e68d097a4a931f4ef901dff0a98082b7e28b7d8

See more details on using hashes here.

File details

Details for the file trustworthyai-0.0.1-py3-none-any.whl.

File metadata

  • Download URL: trustworthyai-0.0.1-py3-none-any.whl
  • Upload date:
  • Size: 153.7 kB
  • Tags: Python 3
  • Uploaded using Trusted Publishing? No
  • Uploaded via: twine/5.1.1 CPython/3.11.10

File hashes

Hashes for trustworthyai-0.0.1-py3-none-any.whl
Algorithm Hash digest
SHA256 bfb1ccde9b3ac8833065f09b7b30efbf1287fa478a207367aedbdf6d22a08a3c
MD5 5b5c6199e41856a59c99b9c3275ae08f
BLAKE2b-256 46e49699f49e9a5d669a6ed0f712067db2db7e25d8cb1305c4d6c439b1c270d9

See more details on using hashes here.

Supported by

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