/PytorchIntegratedGradients

Pytorch implementation of integrate gradients

Primary LanguagePythonBSD 3-Clause "New" or "Revised" LicenseBSD-3-Clause

PytorchIntegratedGradients

Reimplementation of Axiomatic attribution for deep networks in PyTorch.

Results on a lovely cat!

Usage

Start the web server by running

$ python main.py

Then access it by localhost:8000.

Citation

Please cite the original paper by

@inproceedings{sundararajan2017axiomatic,
  title={Axiomatic attribution for deep networks},
  author={Sundararajan, Mukund and Taly, Ankur and Yan, Qiqi},
  booktitle={Proceedings of the 34th International Conference on Machine Learning-Volume 70},
  pages={3319--3328},
  year={2017},
  organization={JMLR.org}
}

Copyright

Mengxiao Lin linmx0130@gmail.com All rights reseved. See LICENSE for more details.