Skip to content

BERT model for Enhancer-Promoter Classification

License

Notifications You must be signed in to change notification settings

smtnkc/bert4epi

Folders and files

NameName
Last commit message
Last commit date

Latest commit

 

History

15 Commits
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 

Repository files navigation

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.

About

BERT model for Enhancer-Promoter Classification

Resources

License

Stars

Watchers

Forks

Releases

No releases published

Packages

No packages published