Based on our paper "A Fuzzy Rank-based Ensemble of CNN Models for Classification of Cervical Cytology" published in Nature- Scientific Reports.
To install the required dependencies run the following in command prompt:
pip install -r requirements.txt
To run the ensemble framework on cervical cytology data, download the data and follow the following directory structure:
(Note: Any number of classes can be present in the dataset. It will be captured by the code automatically)
+-- data
| +-- .
| +-- class1
| +-- class2
| +-- class3
| +-- class4
+-- utils
| +-- .
| +-- utils_cnn
| +-- utils_ensemble
+-- main.py
Then, run the code (5-fold cross-validation will be automatically performed) using the command prompt as follows:
python main.py --data_directory "data/" --epochs 20
If this repository helps you in any way, consider citing our paper as follows:
@article{manna2021fuzzy,
title={A fuzzy rank-based ensemble of CNN models for classification of cervical cytology},
author={Manna, Ankur and Kundu, Rohit and Kaplun, Dmitrii and Sinitca, Aleksandr and Sarkar, Ram},
journal={Scientific Reports},
volume={11},
number={1},
pages={1--18},
year={2021},
publisher={Nature Publishing Group}
}