DerivedFunction commited on
Commit
ce1f225
·
verified ·
1 Parent(s): 3c740a9

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +1 -84
README.md CHANGED
@@ -131,87 +131,4 @@ pipeline_tag: token-classification
131
 
132
 
133
  This model is experimental baseline to test whether or not token classification is viable. It is trained on cleaned sentences, but may have English leakage or biases.
134
- See `polyglot-tagger-v2` and others for the latest version.
135
-
136
- Fine-tuned `xlm-roberta-base` for sentence-level language tagging across 100 languages.
137
- The model predicts BIO-style language tags over tokens, which makes it useful for
138
- language identification, code-switch detection, and multilingual document analysis.
139
-
140
-
141
-
142
- ## Model description (Experimental Version)
143
-
144
- Introducing Polyglot Tagger, a new way to classify multi-lingual documents. By training specifically on token classification on individual sentences, the model
145
- generalizes well on a variety of languages, while also behaves as a multi-label classifier, and extracts sentences based on its language.
146
-
147
- ## Intended uses & limitations
148
- This model can be treated as a base model for further fine-tuning on specific language identification extraction tasks.
149
- Note that as a general language tagging model, it can potentially get confused from shared language families or from short texts. For example, English and German, Spanish and Portuguese, and Russian and Ukrainian.
150
-
151
- The model is trained on a sentence with a minimum of four tokens, so it may not accurately classify very short and ambigous statements. Note that this model is experimental
152
- and may produce unexpected results compared to generic text classifiers. It is trained on cleaned text, therefore, "messy" text may unexpectedly produce different results.
153
-
154
- > Note that Romanized versions of any language is not included in the training set, such as Romanized Russian, and Hindi.
155
-
156
- ### Training and Evaluation Data
157
- A synthetic training row consists of 1-4 individual and mostly independent sentences extracted from various sources. The actual training and evaluation data, as well as coverage
158
- is found in `DerivedFunction/language-ner`.
159
-
160
-
161
- ## Training procedure
162
-
163
- ### Training hyperparameters
164
-
165
- The following hyperparameters were used during training:
166
- - learning_rate: 5e-05
167
- - train_batch_size: 72
168
- - eval_batch_size: 36
169
- - seed: 42
170
- - gradient_accumulation_steps: 2
171
- - total_train_batch_size: 144
172
- - optimizer: Use OptimizerNames.ADAMW_TORCH_FUSED with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
173
- - lr_scheduler_type: linear
174
- - num_epochs: 2
175
- - mixed_precision_training: Native AMP
176
-
177
- ### Training results
178
-
179
- It achieves the following results on the evaluation set:
180
- - Loss: 0.0452
181
- - Precision: 0.8626
182
- - Recall: 0.8916
183
- - F1: 0.8769
184
- - Accuracy: 0.9892
185
-
186
- | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
187
- |:-------------:|:------:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
188
- | 0.0730 | 0.0905 | 2500 | 0.1081 | 0.7241 | 0.8260 | 0.7717 | 0.9760 |
189
- | 0.0622 | 0.1809 | 5000 | 0.1276 | 0.6822 | 0.8122 | 0.7416 | 0.9724 |
190
- | 0.0556 | 0.2714 | 7500 | 0.0826 | 0.7701 | 0.8463 | 0.8064 | 0.9813 |
191
- | 0.0504 | 0.3618 | 10000 | 0.0763 | 0.7916 | 0.8562 | 0.8226 | 0.9822 |
192
- | 0.0480 | 0.4523 | 12500 | 0.0703 | 0.8025 | 0.8602 | 0.8304 | 0.9839 |
193
- | 0.0408 | 0.5427 | 15000 | 0.0750 | 0.8072 | 0.8637 | 0.8345 | 0.9837 |
194
- | 0.0443 | 0.6332 | 17500 | 0.0652 | 0.8149 | 0.8657 | 0.8395 | 0.9849 |
195
- | 0.0403 | 0.7236 | 20000 | 0.0647 | 0.8298 | 0.8728 | 0.8507 | 0.9859 |
196
- | 0.0413 | 0.8141 | 22500 | 0.0590 | 0.8253 | 0.8686 | 0.8464 | 0.9865 |
197
- | 0.0367 | 0.9045 | 25000 | 0.0582 | 0.8288 | 0.8743 | 0.8510 | 0.9867 |
198
- | 0.0395 | 0.9950 | 27500 | 0.0583 | 0.8304 | 0.8768 | 0.8530 | 0.9862 |
199
- | 0.0338 | 1.0854 | 30000 | 0.0567 | 0.8353 | 0.8783 | 0.8562 | 0.9869 |
200
- | 0.0291 | 1.1759 | 32500 | 0.0537 | 0.8443 | 0.8786 | 0.8611 | 0.9878 |
201
- | 0.0300 | 1.2663 | 35000 | 0.0521 | 0.8435 | 0.8805 | 0.8616 | 0.9878 |
202
- | 0.0269 | 1.3568 | 37500 | 0.0531 | 0.8515 | 0.8859 | 0.8683 | 0.9879 |
203
- | 0.0295 | 1.4472 | 40000 | 0.0517 | 0.8548 | 0.8882 | 0.8712 | 0.9882 |
204
- | 0.0279 | 1.5377 | 42500 | 0.0489 | 0.8550 | 0.8884 | 0.8714 | 0.9884 |
205
- | 0.0281 | 1.6281 | 45000 | 0.0480 | 0.8551 | 0.8875 | 0.8710 | 0.9887 |
206
- | 0.0277 | 1.7186 | 47500 | 0.0467 | 0.8605 | 0.8904 | 0.8752 | 0.9888 |
207
- | 0.0289 | 1.8090 | 50000 | 0.0458 | 0.8599 | 0.8919 | 0.8756 | 0.9892 |
208
- | 0.0268 | 1.8995 | 52500 | 0.0457 | 0.8623 | 0.8906 | 0.8762 | 0.9891 |
209
- | 0.0306 | 1.9899 | 55000 | 0.0452 | 0.8626 | 0.8916 | 0.8769 | 0.9892 |
210
-
211
-
212
- ### Framework versions
213
-
214
- - Transformers 5.0.0
215
- - Pytorch 2.10.0+cu128
216
- - Datasets 4.0.0
217
- - Tokenizers 0.22.2
 
131
 
132
 
133
  This model is experimental baseline to test whether or not token classification is viable. It is trained on cleaned sentences, but may have English leakage or biases.
134
+ See `polyglot-tagger-v2` and others for the latest version.