Skip to content

Latest commit

 

History

History
27 lines (25 loc) · 1.16 KB

README.md

File metadata and controls

27 lines (25 loc) · 1.16 KB
Raw

RuleBERT + bAbi

This file shows how to apply RuleBERT on bAbi Task #15.

Recipe

  1. Download RuleBERT Model
bash download_model.sh
  1. Download Data The dataset can be downloaded here. We are interested in Task #15. Copy the files from en_valid/ to data/external_datasets/bAbI/.
  2. Fine-tune on bAbI
#RuleBERT
!python experiments/external_datasets/bAbI/run_bAbI.py --babi_dir "data/external_datasets/bAbI/" \
                                                       --model_arch 'models/rulebert_161' \
                                                       --epochs 3 \
                                                       --reps 10
#Roberta
!python experiments/external_datasets/bAbI/run_bAbI.py --babi_dir "data/external_datasets/bAbI/" \
                                                       --model_arch 'roberta-large' \
                                                       --epochs 3 \
                                                       --reps 10

The results are stored in --babi_dir.

License

MIT