update model card README.md

README.md

@@ -18,19 +18,12 @@ should probably proofread and complete it, then remove this comment. -->
 # bert-finetuned-gesture-prediction-21-classes
 
 This model is a fine-tuned version of [bert-base-cased](https://huggingface.co/bert-base-cased) on the None dataset.
-It achieves the following results on the validation set:
-- Loss: 0.8967
-- Accuracy:  0.8014
-- Precision: 0.7931
-- Recall: 0.8014
-- F1: 0.7913
-
-It achieves the following results on the test set:
-- Loss: 0.8763
-- Accuracy:  0.7896
-- Precision: 0.7867
-- Recall: 0.7896
-- F1: 0.7808
+It achieves the following results on the evaluation set:
+- Loss: 0.9660
+- Accuracy: 0.8214
+- Precision: 0.8184
+- Recall: 0.8214
+- F1: 0.8166
 
 ## Model description
 
@@ -42,7 +35,7 @@ More information needed
 
 ## Training and evaluation data
 
-The model has been trained, validated, and tested with the qfrodicio/gesture-prediction-21-classes dataset
+More information needed
 
 ## Training procedure
 
@@ -50,7 +43,6 @@ The model has been trained, validated, and tested with the qfrodicio/gesture-pre
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- weight_decay: 0.01
 - train_batch_size: 16
 - eval_batch_size: 16
 - seed: 42
@@ -62,16 +54,16 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | Precision | Recall | F1     |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:---------:|:------:|:------:|
-| 2.1826        | 1.0   | 104  | 1.2892          | 0.7193   | 0.6656    | 0.7193 | 0.6745 |
-| 1.0758        | 2.0   | 208  | 0.9593          | 0.7816   | 0.7519    | 0.7816 | 0.7621 |
-| 0.6947        | 3.0   | 312  | 0.9170          | 0.7885   | 0.7730    | 0.7885 | 0.7760 |
-| 0.4928        | 4.0   | 416  | 0.8967          | 0.8014   | 0.7931    | 0.8014 | 0.7913 |
-| 0.3413        | 5.0   | 520  | 0.9195          | 0.8034   | 0.8076    | 0.8034 | 0.7966 |
-| 0.2473        | 6.0   | 624  | 0.9123          | 0.8131   | 0.8134    | 0.8131 | 0.8078 |
-| 0.1799        | 7.0   | 728  | 0.9474          | 0.8191   | 0.8195    | 0.8191 | 0.8139 |
-| 0.1329        | 8.0   | 832  | 0.9611          | 0.8237   | 0.8222    | 0.8237 | 0.8182 |
-| 0.116         | 9.0   | 936  | 0.9849          | 0.8204   | 0.8190    | 0.8204 | 0.8148 |
-| 0.0972        | 10.0  | 1040 | 0.9818          | 0.8232   | 0.8213    | 0.8232 | 0.8181 |
+| 2.225         | 1.0   | 104  | 1.3314          | 0.7115   | 0.6469    | 0.7115 | 0.6675 |
+| 1.0881        | 2.0   | 208  | 0.9569          | 0.7750   | 0.7577    | 0.7750 | 0.7525 |
+| 0.7006        | 3.0   | 312  | 0.8805          | 0.7959   | 0.7917    | 0.7959 | 0.7831 |
+| 0.4943        | 4.0   | 416  | 0.8664          | 0.8123   | 0.8122    | 0.8123 | 0.8048 |
+| 0.3372        | 5.0   | 520  | 0.8765          | 0.8130   | 0.8102    | 0.8130 | 0.8053 |
+| 0.2416        | 6.0   | 624  | 0.8772          | 0.8166   | 0.8139    | 0.8166 | 0.8107 |
+| 0.178         | 7.0   | 728  | 0.9186          | 0.8217   | 0.8186    | 0.8217 | 0.8167 |
+| 0.1302        | 8.0   | 832  | 0.9186          | 0.8202   | 0.8183    | 0.8202 | 0.8165 |
+| 0.1063        | 9.0   | 936  | 0.9618          | 0.8245   | 0.8213    | 0.8245 | 0.8198 |
+| 0.094         | 10.0  | 1040 | 0.9660          | 0.8214   | 0.8184    | 0.8214 | 0.8166 |
 
 
 ### Framework versions