End of training

README.md

@@ -6,6 +6,9 @@ tags:
 - generated_from_trainer
 metrics:
 - accuracy
+- recall
+- precision
+- f1
 model-index:
 - name: distilbert-base-uncased-finetuned-text_classification
   results: []
@@ -18,8 +21,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.7908
-- Accuracy: 0.8699
+- Loss: 0.8070
+- Accuracy: 0.8804
+- Recall: 0.9572
+- Precision: 0.8557
+- F1: 0.9036
 
 ## Model description
 
@@ -48,18 +54,18 @@ The following hyperparameters were used during training:
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:--------:|
-| No log        | 1.0   | 231  | 0.3741          | 0.7900   |
-| No log        | 2.0   | 462  | 0.4525          | 0.8168   |
-| 0.2312        | 3.0   | 693  | 0.4493          | 0.8635   |
-| 0.2312        | 4.0   | 924  | 0.5794          | 0.8559   |
-| 0.0811        | 5.0   | 1155 | 0.6408          | 0.8559   |
-| 0.0811        | 6.0   | 1386 | 0.6693          | 0.8705   |
-| 0.0222        | 7.0   | 1617 | 0.7076          | 0.8786   |
-| 0.0222        | 8.0   | 1848 | 0.7893          | 0.8693   |
-| 0.008         | 9.0   | 2079 | 0.7670          | 0.8734   |
-| 0.008         | 10.0  | 2310 | 0.7908          | 0.8699   |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | Recall | Precision | F1     |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
+| No log        | 1.0   | 231  | 0.3988          | 0.8460   | 0.9462 | 0.8190    | 0.8780 |
+| No log        | 2.0   | 462  | 0.3742          | 0.8757   | 0.9432 | 0.8586    | 0.8989 |
+| 0.148         | 3.0   | 693  | 0.5030          | 0.8670   | 0.9323 | 0.8540    | 0.8914 |
+| 0.148         | 4.0   | 924  | 0.6576          | 0.8699   | 0.9592 | 0.8410    | 0.8962 |
+| 0.0525        | 5.0   | 1155 | 0.7022          | 0.8641   | 0.9552 | 0.8361    | 0.8917 |
+| 0.0525        | 6.0   | 1386 | 0.8569          | 0.8617   | 0.9671 | 0.8264    | 0.8912 |
+| 0.0148        | 7.0   | 1617 | 0.7352          | 0.8769   | 0.9542 | 0.8531    | 0.9008 |
+| 0.0148        | 8.0   | 1848 | 0.8513          | 0.8705   | 0.9612 | 0.8406    | 0.8968 |
+| 0.0075        | 9.0   | 2079 | 0.7685          | 0.8781   | 0.9482 | 0.8584    | 0.9011 |
+| 0.0075        | 10.0  | 2310 | 0.8070          | 0.8804   | 0.9572 | 0.8557    | 0.9036 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:8c6af388f58f8202133d2ab385eae63b0d4901710cf3bb0b3047e43d51667c0c
+oid sha256:91392685c74cca23a5b916fdd44674109494da320aee5361aae798080506fd7a
 size 267832560