Skip to content

Commit 33dd2a1

Browse files
committed
This model seems to work slightly better, so set it to be the default
1 parent 3c6d2a1 commit 33dd2a1

File tree

1 file changed

+3
-1
lines changed

1 file changed

+3
-1
lines changed

stanza/resources/default_packages.py

+3-1
Original file line numberDiff line numberDiff line change
@@ -791,9 +791,10 @@ def build_default_pretrains(default_treebanks):
791791
# hfl bert: 0.9469
792792
# hfl roberta: 0.9459
793793
# hfl electra: 0.9515
794+
# hfl macbert: 0.9530
794795
# There is also a ShannonAI model, but our current codebase is
795796
# somehow not compatible
796-
"zh-hans": "hfl/chinese-electra-180g-large-discriminator",
797+
"zh-hans": "hfl/chinese-macbert-large",
797798
}
798799

799800
TRANSFORMER_LAYERS = {
@@ -894,6 +895,7 @@ def build_default_pretrains(default_treebanks):
894895
# zh
895896
"google-bert/bert-base-chinese": "google-bert-chinese",
896897
"hfl/chinese-bert-wwm": "hfl-bert-chinese",
898+
"hfl/chinese-macbert-large": "hfl-macbert-chinese",
897899
"hfl/chinese-roberta-wwm-ext": "hfl-roberta-chinese",
898900
"hfl/chinese-electra-180g-large-discriminator": "electra-large",
899901
"ShannonAI/ChineseBERT-base": "shannonai-chinese-bert",

0 commit comments

Comments
 (0)