Skip to content

Commit 534e5c9

Browse files
committed
Add comments on BG models and set a default BG transformer
1 parent 9365aaa commit 534e5c9

File tree

1 file changed

+10
-0
lines changed

1 file changed

+10
-0
lines changed

stanza/resources/default_packages.py

Lines changed: 10 additions & 0 deletions
Original file line numberDiff line numberDiff line change
@@ -473,6 +473,15 @@ def build_default_pretrains(default_treebanks):
473473
# aubmindlab/araelectra-base-discriminator 95.66 85.83 86.10
474474
"ar": "aubmindlab/araelectra-base-discriminator",
475475

476+
# Tried a few different BG models on BTB depparse
477+
# bg_btb LAS dev test
478+
# no transformer 90.19 91.26
479+
# MaCoCu/BERTovski 90.88 91.74
480+
# rmihaylov/bert-base-theseus-bg 91.81 93.21
481+
# DeepPavlov/bert-base-bg-cs-pl-ru-cased 92.17 93.47
482+
# rmihaylov/bert-base-bg 92.78 94.09
483+
"bg": "rmihaylov/bert-base-bg",
484+
476485
# https://huggingface.co/Maltehb/danish-bert-botxo
477486
# contrary to normal expectations, this hurts F1
478487
# on a dev split by about 1 F1
@@ -865,6 +874,7 @@ def build_default_pretrains(default_treebanks):
865874
"aubmindlab/bert-base-arabertv2": "aubmind-bert",
866875

867876
# bg
877+
# these two required hacks to get the model to function (see bert_model.py)
868878
"rmihaylov/bert-base-bg": "bert-base-bg",
869879
"rmihaylov/bert-base-theseus-bg": "bert-base-theseus-bg",
870880
# may also work on Macedonian? the Macedonian UD is not large enough for building models, though

0 commit comments

Comments
 (0)