Metadata-Version: 2.1
Name: nam-pytorch
Version: 0.0.1
Summary: Neural Additive Models (NAM) - Pytorch
Home-page: https://github.com/rish-16/nam-pytorch
Author: Rishabh Anand
Author-email: mail.rishabh.anand@gmail.com
License: MIT
Keywords: artificial intelligence,deep learning,nam,neural additive models,generalized additive models
Platform: UNKNOWN
Classifier: Development Status :: 4 - Beta
Classifier: Intended Audience :: Developers
Classifier: Topic :: Scientific/Engineering :: Artificial Intelligence
Classifier: License :: OSI Approved :: MIT License
Classifier: Programming Language :: Python :: 3.6
Description-Content-Type: text/markdown
Requires-Dist: torch (>=1.6)

# nam-pytorch
Unofficial PyTorch implementation of Neural Additive Models (NAM) by Agarwal, et al. [[`abs`](https://arxiv.org/abs/2004.13912), [`pdf`](https://arxiv.org/pdf/2004.13912.pdf)]

<img width=600 src="https://github.com/rish-16/nam-pytorch/raw/main/pic.png" />

---

## Installation

You can access `nam-pytorch` via `pip`:

```bash
$ pip install nam-pytorch
```

## Usage

```python
import torch 
from nam_pytorch import NAM

nam = NAM(
    num_features=784,
    link_func="sigmoid"
)

images = torch.rand(32, 784)
pred = nam(images) # [32, 1]
```

## Contributing
As always, if there are any issues with / suggestions for the code, feel free to raise an issue or submit a PR.

## License
[MIT](https://github.com/rish-16/nam-pytorch/blob/main/LICENSE)


