Neural Additive Models (NAM) - Pytorch
Project description
nam-pytorch
Unofficial PyTorch implementation of Neural Additive Models (NAM) by Agarwal, et al. [abs, pdf]
Installation
You can access nam-pytorch via pip:
$ pip install nam-pytorch
Usage
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
Release history Release notifications | RSS feed
Download files
Download the file for your platform. If you're not sure which to choose, learn more about installing packages.
Source Distribution
nam_pytorch-0.0.1.tar.gz
(2.2 kB
view details)
Built Distribution
Filter files by name, interpreter, ABI, and platform.
If you're not sure about the file name format, learn more about wheel file names.
Copy a direct link to the current filters
File details
Details for the file nam_pytorch-0.0.1.tar.gz.
File metadata
- Download URL: nam_pytorch-0.0.1.tar.gz
- Upload date:
- Size: 2.2 kB
- Tags: Source
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.1 importlib_metadata/3.7.2 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.60.0 CPython/3.6.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
8ea2ccb7c313daa1690fcfb74ef074c321722453028ab51cf6df5ee4664e0090
|
|
| MD5 |
4808f8972beeaff9f714ee9cb44b268d
|
|
| BLAKE2b-256 |
f0ace727439b3cbb4d911b53385d190a97bf6e93f6ed8ec587ebeea93b70f8f1
|
File details
Details for the file nam_pytorch-0.0.1-py3-none-any.whl.
File metadata
- Download URL: nam_pytorch-0.0.1-py3-none-any.whl
- Upload date:
- Size: 3.5 kB
- Tags: Python 3
- Uploaded using Trusted Publishing? No
- Uploaded via: twine/3.4.1 importlib_metadata/3.7.2 pkginfo/1.7.0 requests/2.25.1 requests-toolbelt/0.9.1 tqdm/4.60.0 CPython/3.6.13
File hashes
| Algorithm | Hash digest | |
|---|---|---|
| SHA256 |
bc611b1f785cdd50a5ebc6356e2b9f611514b89618d2684ff00ce2becf3c0574
|
|
| MD5 |
8da39e074212bd991cf9283bb8d1ffd5
|
|
| BLAKE2b-256 |
84fb5028751f4817e77221d7dd1deb56a09fbf430d36739c526bac0838bd1919
|
