This folder contains the source code to run DRAIN method.

#################################

To run experiment on 2-Moons dataset, go to classification folder and do

1. name model_moons.py as model.py

2. python train.py --dataset Moons


Similar process for other datasets.

#################################

The code has been tested with PyTorch and Anaconda.