# Antibody Specificity Classification I have already added other models. So, please pull the updated version before running these following commands. ``` git pull ``` ## HuggingFace ### SciBERT With the updated version, there is no need to download any models, you can just run this following command: ``` python train_k_fold_cross_val.py --pretrained_bert_name allenai/scibert_scivocab_uncased ``` ### Other BERT models Similarly, for other BERT models supported by [huggingface](https://huggingface.co/transformers/), you can simply modified `--pretranined_bert_name [other BERT models]` to use those models. For example: ``` python train_k_fold_cross_val.py --pretrained_bert_name roberta-base ``` or for bioBERT: ``` python train_k_fold_cross_val.py --pretrained_bert_name dmis-lab/biobert-v1.1 ``` ## Other models ### List of available models * bert_spc * aoa_bert * fc_aoa_bert * bi_aoa_bert * mha_aoa_bert * aoa_cls_bert * bi_aoa_cls_bert * mha_aoa_cls_bert * aen_bert * aen_cls_bert * lcf_bert * lcf_cls_bert ### To run these models ```bash python train_k_fold_cross_val.py --model_name [model_name] ```