/iDNA_ABT

python codes for iDNA-ABT:advanced deep learning model for detecting DNA methylation with adaptive features and transductive information maximization

Primary LanguagePython

iDNA_ABT

Check environment

you can run both on cpu and gpu.

Environment: pytorch; pynvml; pickle; numpy; matplotlib; seaborn; pandas.

After processing, you can run env_test/version_test.py and env_test/gpu_test.py.

Train process

  1. You can change model or training parameters in configuration/config.py to train models.

  2. Change dataset to yours if you need, and we support defalut dataset in data/DNA_MS.

    python train/main.py

Predict process

Specify pt and config path, and python train/inference.py.

Other: