bert4epi

BERT model for Enhancer-Promoter Classification

python3 prepare_data.py --k_fold=5 --balanced
python3 train.py --cell_line='GM12878' --k_fold=5
python3 test.py --cell_line='GM12878' --cross_cell_line='K562' --k_fold=5

⚠️ By default --seed=42.

⚠️ Set --cell_line and --cross_cell_line as same for testing on the same cell-line.

⚠️ Set --k_fold=0 to disable cross-validation.