README.md

PyTorch CBC

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

Google Colab demo:

link

Install requirements

pip install -r requirements.txt

To train MNIST

python mnist.py