rashley-iqt / aiscan

Scan your AI/ML models for problems.

Geek Repo:Geek Repo

Github PK Tool:Github PK Tool

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.

About

Scan your AI/ML models for problems.

License:Apache License 2.0


Languages

Language:Python 100.0%