Probabilistic Integral Circuits - PICs
This repository is the official implementation of Probabilistic Integral Circuits.
Datasets
-
MNIST-famility datasets: all MNIST-famility datasets are available in torchvision, and automatically downloaded if needed.
-
PTB288: download it here, and place it in
data/ptbchar_288
-
Binary datasets (DEBD): download them here, and place them in
data/debd
-
UCI datasets: download non pre-processed datasets here or pre-processed datasets here, and place them in
data/UCI
Training PICs
python train_pic.py -ds mnist -bs 256 -nip 128 -int trapezoidal
python train_pic.py -ds fashion_mnist -bs 256 -nip 128 -int trapezoidal
python train_pic.py -ds emnist -split mnist -bs 256 -nip 128 -int trapezoidal
python train_pic.py -ds emnist -split letters -bs 256 -nip 128 -int trapezoidal
python train_pic.py -ds emnist -split balanced -bs 256 -nip 128 -int trapezoidal
python train_pic.py -ds emnist -split byclass -bs 256 -nip 128 -int trapezoidal
Training HCLTs
python train_hclt.py -ds mnist -bs 256 -hd 128
python train_hclt.py -ds fashion_mnist -bs 256 -hd 128
python train_hclt.py -ds emnist -split mnist -bs 256 -hd 128
python train_hclt.py -ds emnist -split letters -bs 256 -hd 128
python train_hclt.py -ds emnist -split balanced -bs 256 -hd 128
python train_hclt.py -ds emnist -split byclass -bs 256 -hd 128