End of training

README.md

@@ -3,6 +3,11 @@ license: apache-2.0
 base_model: distilbert/distilbert-base-multilingual-cased
 tags:
 - generated_from_trainer
+metrics:
+- accuracy
+- f1
+- precision
+- recall
 model-index:
 - name: distilbert-lang-detect
   results: []
@@ -14,6 +19,12 @@ should probably proofread and complete it, then remove this comment. -->
 # distilbert-lang-detect
 
 This model is a fine-tuned version of [distilbert/distilbert-base-multilingual-cased](https://huggingface.co/distilbert/distilbert-base-multilingual-cased) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.1215
+- Accuracy: 0.9720
+- F1: 0.9720
+- Precision: 0.9720
+- Recall: 0.9720
 
 ## Model description
 
@@ -41,6 +52,17 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 5
 
+### Training results
+
+| Training Loss | Epoch | Step  | Validation Loss | Accuracy | F1     | Precision | Recall |
+|:-------------:|:-----:|:-----:|:---------------:|:--------:|:------:|:---------:|:------:|
+| 0.1022        | 1.0   | 14547 | 0.0924          | 0.9703   | 0.9703 | 0.9703    | 0.9703 |
+| 0.0783        | 2.0   | 29094 | 0.0924          | 0.9704   | 0.9704 | 0.9705    | 0.9704 |
+| 0.0646        | 3.0   | 43641 | 0.1096          | 0.9724   | 0.9724 | 0.9725    | 0.9724 |
+| 0.053         | 4.0   | 58188 | 0.1188          | 0.9722   | 0.9722 | 0.9722    | 0.9722 |
+| 0.0364        | 5.0   | 72735 | 0.1215          | 0.9720   | 0.9720 | 0.9720    | 0.9720 |
+
+
 ### Framework versions
 
 - Transformers 4.39.2

config.json

@@ -25,6 +25,7 @@
   "n_layers": 6,
   "output_past": true,
   "pad_token_id": 0,
+  "problem_type": "single_label_classification",
   "qa_dropout": 0.1,
   "seq_classif_dropout": 0.2,
   "sinusoidal_pos_embds": false,

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:14bfddc090d5a432146096b244167c93b59fc3b66fc77f521c9d8efc21aa46ff
+oid sha256:44b91e96f191566e36617588ebf9b08805b0babbdb02a9de92d071133df76eef
 size 541320452