End of training

README.md

@@ -4,7 +4,9 @@ base_model: google-bert/bert-base-uncased
 tags:
 - generated_from_trainer
 metrics:
-- accuracy
+- precision
+- recall
+- f1
 model-index:
 - name: Aroma-classifier
   results: []
@@ -15,10 +17,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 # Aroma-classifier
 
-This model is a fine-tuned version of [google-bert/bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased) on an unknown dataset.
+This model is a fine-tuned version of [google-bert/bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.7545
-- Accuracy: 0.4
+- Loss: 0.7405
+- Precision: 0.8273
+- Recall: 0.8268
+- F1: 0.8270
 
 ## Model description
 
@@ -47,11 +51,11 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 2    | 0.8221          | 0.4      |
-| No log        | 2.0   | 4    | 0.8166          | 0.4      |
-| No log        | 3.0   | 6    | 0.7545          | 0.4      |
+| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     |
+|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|
+| 0.4369        | 1.0   | 2375 | 0.5383          | 0.8841    | 0.6854 | 0.7722 |
+| 0.3339        | 2.0   | 4750 | 0.5662          | 0.8410    | 0.8102 | 0.8253 |
+| 0.2333        | 3.0   | 7125 | 0.7405          | 0.8273    | 0.8268 | 0.8270 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:2e8cf59365a1d50379a46c1d38c97ba3453c1c8b635b13f884ee6983e8dabd3e
+oid sha256:06624eb1c8b0a44d3925ce74e236f2b706e1c645fd314b0ab2a618ed29982a66
 size 437958648

runs/Apr05_12-19-55_c2bf071554f1/events.out.tfevents.1712319613.c2bf071554f1.34.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:341071048e8ab82e66697b8360f8c20c5c2db4f7fa7225fbb972f38b006f7060
-size 8373
+oid sha256:f4d7c6cb082da6842a98a70f64fdf0b681862f89885320645f6c61f5995df708
+size 9147