system HF staff commited on
Commit
827846e
1 Parent(s): 4d394c8

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +9 -7
README.md CHANGED
@@ -1,6 +1,6 @@
1
  # dv-wave
2
 
3
- This is a second attempt at a Dhivehi language model trained with
4
  Google Research's [ELECTRA](https://github.com/google-research/electra).
5
 
6
  Tokenization and pre-training CoLab: https://colab.research.google.com/drive/1ZJ3tU9MwyWj6UtQ-8G7QJKTn-hG1uQ9v?usp=sharing
@@ -10,17 +10,19 @@ Using SimpleTransformers to classify news https://colab.research.google.com/driv
10
  V1: similar performance to mBERT on news classification task after finetuning for 3 epochs (52%)
11
 
12
  V2: fixed tokenizers do_lower_case=False and strip_accents=False to preserve vowel signs of Dhivehi
13
- dv-wave: 89% to mBERT: 52%
 
 
 
 
 
14
 
15
  ## Corpus
16
 
17
  Trained on @Sofwath's 307MB corpus of Dhivehi text: https://github.com/Sofwath/DhivehiDatasets
18
 
19
- This repo also contains the news classification task CSV
20
-
21
- [OSCAR](https://oscar-corpus.com/) was considered but has not been added to pretraining; as of
22
- this writing their web crawl has 126MB of Dhivehi text (79MB deduped).
23
 
24
  ## Vocabulary
25
 
26
- Included as vocab.txt in the upload - vocab_size is 29874
 
1
  # dv-wave
2
 
3
+ This is my third version of a Dhivehi language model trained with
4
  Google Research's [ELECTRA](https://github.com/google-research/electra).
5
 
6
  Tokenization and pre-training CoLab: https://colab.research.google.com/drive/1ZJ3tU9MwyWj6UtQ-8G7QJKTn-hG1uQ9v?usp=sharing
 
10
  V1: similar performance to mBERT on news classification task after finetuning for 3 epochs (52%)
11
 
12
  V2: fixed tokenizers do_lower_case=False and strip_accents=False to preserve vowel signs of Dhivehi
13
+
14
+ > 8-topic news classification score: 88.6% compared to mBERT: 51.8%
15
+
16
+ V3: trained longer on larger corpus (added OSCAR and Wikipedia)
17
+
18
+ > news classification score: 89.2%
19
 
20
  ## Corpus
21
 
22
  Trained on @Sofwath's 307MB corpus of Dhivehi text: https://github.com/Sofwath/DhivehiDatasets
23
 
24
+ Plus [OSCAR](https://oscar-corpus.com/) and 2020-10-01 dump of dv.wikipedia.org
 
 
 
25
 
26
  ## Vocabulary
27
 
28
+ Included as vocab.txt in the upload