saralajew / torch-cbc Public

forked from EmilioOldenziel/torch-cbc

Notifications You must be signed in to change notification settings
Fork 0
Star 0

A PyTorch implementation of the paper "Classification-by-Components: Modeling Probabilistic Reasoning over a Set of Components"

0 stars 5 forks Branches Tags Activity

Notifications

Name		Name	Last commit message	Last commit date
Latest commit History 48 Commits
torch_cbc		torch_cbc
.gitignore		.gitignore
README.md		README.md
cbc.gif		cbc.gif
cifar10.py		cifar10.py
mnist.py		mnist.py
mnist_cnn.pt		mnist_cnn.pt
requirements.txt		requirements.txt
utils.py		utils.py

Repository files navigation

PyTorch CBC

A PyTorch implementation of the NeurIPS 2019 paper "Classification-by-Components: Modeling Probabilistic Reasoning over a Set of Components"

Google Colab demo:

Install requirements

pip install -r requirements.txt

To train MNIST

python mnist.py

About

A PyTorch implementation of the paper "Classification-by-Components: Modeling Probabilistic Reasoning over a Set of Components"

Report repository

Releases

No releases published

Packages

No packages published

Languages

Python 100.0%