Soft Decision Tree classifier: explainable and extendable PyTorch implementation

Reuben R Shamir

We implemented a Soft Decision Tree (SDT) and a Short-term Memory Soft Decision Tree (SM-SDT) using PyTorch. The methods were extensively tested on simulated and clinical datasets.…