bert-syntax

by yoavg

Assessing syntactic abilities of BERT

135 Stars 18 Forks Last release: Not found Apache License 2.0 3 Commits 0 Releases

Available items

No Items, yet!

The developer of this repository has not created any items for sale yet. Need a bug fixed? Help with integration? A different license? Create a request here:

BERT-Syntax

Assesing the syntactic abilities of BERT.

What

Evaluate Google's BERT-Base and BERT-Large models on the syntactic agreement datasets from Linzen, Goldberg and Dupoux 2016 and Marvin and Linzen 2018 and Gulordava et al 2018.

This is quite messy, as I hacked it together between things here and there. But I also believe it is accurate. This lists the data files and shows how to run the evaluation. For more details and results, see the arxiv report.

Data Files

Data taken from the github repos of Linzen, Goldberg and Dupoux (LGD), Marvin and Linzen (ML), and Gulordava et al.

| File | Description | |---|---| |

marvin_linzen_dataset.tsv
| stimuli from Marvin and Linzen. I dumped it from the pickle files of ML | |
wiki.vocab
| from LGD, used for verb inflections (wiki.vocab) | |
lgd_dataset.tsv
| processed data from LGD | |
generated.tab
| data from Gulordava et al (generated.tab) |

lgd_dataset.tsv
is created by
bash
wget http://tallinzen.net/media/rnn_agreement/agr_50_mostcommon_10K.tsv.gz
gunzip agr_50_mostcommon_10K.tsv.gz
python make_linzen_goldberg_testset.py > lgd_dataset.tsv

Obtaining the results

pip install pytorch_pretrained_bert

python eval_bert.py > results/lgd_results_large.txt python eval_bert.py base > results/lgd_results_base.txt python eval_bert.py marvin > results/marvin_results_large.txt python eval_bert.py marvin base > results/marvin_results_base.txt python eval_bert.py gul > results/gulordava_results_large.txt python eval_bert.py gul base > results/gulordava_results_base.txt

Generating tables (for the PDF)

python gen_marvin_tbl.py 
python gen_lgd_tbl.py
python gen_gul_tbl.py

We use cookies. If you continue to browse the site, you agree to the use of cookies. For more information on our use of cookies please see our Privacy Policy.