Any question is welcomed to be asked by issue and I will try my best to solve your problems.
You can change parameters in configuration/config.py
to train models.
You can change model structure in model/ClassificationDNAbert.py
to train models.
You can change training process and dataset process in frame/ModelManager.py
and frame/DataManager.py
.
Besides, dataset in paper is also included in data/DNA_MS
.
You should download pretrain model from relevant github repository.
For example, if you want to use DNAbert, you need to put them into the pretrain folder and rename the relevant choice in the model.
python main/train.py