DuongTrongChi/facebook-commet-classification-small

README.md

@@ -1,4 +1,5 @@
 ---
+base_model: uitnlp/visobert
 tags:
 - generated_from_trainer
 metrics:
@@ -6,7 +7,6 @@ metrics:
 - f1
 - precision
 - recall
-base_model: uitnlp/visobert
 model-index:
 - name: facebook-commet-classification-small
   results: []
@@ -17,13 +17,13 @@ should probably proofread and complete it, then remove this comment. -->
 
 # facebook-commet-classification-small
 
-This model is a fine-tuned version of [uitnlp/visobert](https://huggingface.co/uitnlp/visobert) on the None dataset.
+This model is a fine-tuned version of [uitnlp/visobert](https://huggingface.co/uitnlp/visobert) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.1327
-- Accuracy: 0.9578
-- F1: 0.7406
-- Precision: 0.8167
-- Recall: 0.6774
+- Loss: 0.1605
+- Accuracy: 0.9427
+- F1: 0.7552
+- Precision: 0.8298
+- Recall: 0.6929
 
 ## Model description
 
@@ -54,7 +54,7 @@ The following hyperparameters were used during training:
 
 | Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
 |:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
-| 0.1282        | 1.0   | 1828 | 0.1327          | 0.9578   | 0.7406 | 0.8167    | 0.6774 |
+| 0.2261        | 1.0   | 990  | 0.1605          | 0.9427   | 0.7552 | 0.8298    | 0.6929 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:17f37f007eae82f056569092ddf910b2682b464768c769b855369c473328395d
+oid sha256:8cef57604a5447f9b0268cd8700e26a81e615cd3f8f6077baa6aa0dfc6f7383d
 size 390300192

runs/Apr11_05-21-47_4b0604d42325/events.out.tfevents.1712812908.4b0604d42325.1400.0

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:44f3ef24e971107e935bce8c7fb83386034ff9bc48948a1a747ba7af187c0190
+size 6001

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:fc3b7bcbcff825940a23b6dd6472bfb63dd9d6c0c266e224a398efdb0024a3ac
+oid sha256:1ba531edc2c6155662e4f83f25024e3d7e400c4988e8e93e9758119d0336f370
 size 4920