Skip to content

Latest commit

 

History

History
13 lines (10 loc) · 429 Bytes

README.md

File metadata and controls

13 lines (10 loc) · 429 Bytes

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.