GAMI-Net Pytorch version

Generalized additive models with structured interactions - PyTorch version

Installation

pip install git+https://github.com/SelfExplainML/GamiNet-PyTorch

Citations

@article{yang2021gami,
   title={GAMI-Net: An Explainable Neural Network based on Generalized Additive Models with Structured Interactions},
   author={Yang, Zebin and Zhang, Aijun and Sudjianto, Agus},
   journal={Pattern Recognition},
   volume = {120},
   pages = {108192},
   year={2021}
}