Panda0116/emotion-classification-model

README.md

@@ -7,19 +7,19 @@ tags:
 metrics:
 - accuracy
 model-index:
-- name: results
+- name: emotion-classification-model
   results: []
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
 should probably proofread and complete it, then remove this comment. -->
 
-# results
+# emotion-classification-model
 
 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.1791
-- Accuracy: 0.925
+- Loss: 0.1606
+- Accuracy: 0.9405
 
 ## Model description
 
@@ -51,9 +51,9 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|
-| 0.2315        | 1.0   | 1000 | 0.1942          | 0.9235   |
-| 0.1279        | 2.0   | 2000 | 0.1868          | 0.933    |
-| 0.0818        | 3.0   | 3000 | 0.1670          | 0.938    |
+| 0.2293        | 1.0   | 1000 | 0.1746          | 0.9335   |
+| 0.1315        | 2.0   | 2000 | 0.1529          | 0.937    |
+| 0.0798        | 3.0   | 3000 | 0.1606          | 0.9405   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:0b101eb92f63abca6ef9d90109b13bd918b4ce6375c39119597888d0bf467c7f
+oid sha256:86dc8318e81cd9efba3ae8d93965998a9c492a50a0806a3ae9c3e5029859ab13
 size 267844872