This repository contains an implementation of CPT The code was tested for python 3.6
You can use the following command to train CPT for the three depth paramters of 8, 9, and 10:
python run_polytop.py --data MNIST --depth 8 9 10 --lr 0.01 --epochs 70 --refine_epochs 50 -s 0.01 --initial_steepness 1 --polytope_sides 60 --refine
And for Connect4, depth 2
python run_polytop.py --data connect4 --depth 2 --lr 0.01 --epochs 40 --refine_epochs 40 -s 0.01 --initial_steepness 1 --polytope_sides 40 --refine
For regression use the file run_polytop_reg.pyand for binary classification run the file run_polytop_binary_classification.py.
The datasets are also provided.