Skip to main content

Scan AI models for problems

Project description

AI Scan

This is a scanner that will scan your AI models for problems. Currently it focuses on bias testing. It is currently pre-alpha.

How to Use

First install it:

pip install aiscan

Run it in this manner (currently supports models from HuggingFace’s repository):

aiscan --huggingface roberta-base --task fill-mask

Here is another example with a different bias task.

aiscan --huggingface cross-encoder/nli-distilroberta-base --task zero-shot-classification

That’s it for now. More will come.

Future Work

  • More bias tests. More metrics for bias testing based on the research in the field.

  • Integration with other types of testing (eg. adversarial robustness)

  • More kinds of models besides HuggingFace models. We are especially interested in MLFlow integration.

  • Documentation.

Please contribute if you can. Help is always helpful.

License

Apache

Credit

A project of IQT Labs.

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

aiscan-0.1.0.tar.gz (4.1 kB view hashes)

Uploaded Source

Built Distribution

aiscan-0.1.0-py2.py3-none-any.whl (8.6 kB view hashes)

Uploaded Python 2 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