GitHub repository: xuyxu/Soft-Decision-Tree
Primary language: Python
Description: PyTorch Implementation of "Distilling a Neural Network Into a Soft Decision Tree." Nicholas Frosst, Geoffrey Hinton., 2017.