python version : 3.10
Requirements:
- numpy
- scikit-learn
- torch
- torchvision
- scipy
- PIL
- visdom
- pandas
- pingouin
All the benchmark datasets are uploaded to datasets/Benchmark directory. Run following command to test RGP on these datasets:
python benchmark.py
You can also modify the DATASETS array in benchmark.py to perform specific experiments on different benchmark datasets.
Run following command to test DSMIL on these datasets for comparison:
python train_mil.py
You can also modify the DATASETS array in train_mil.py to perform specific experiments on different benchmark datasets.