sungjin228/GoEmotions_FT_2

README.md

@@ -26,16 +26,16 @@ model-index:
     metrics:
     - name: F1
       type: f1
-      value: 0.7207163601161665
+      value: 0.7211585665193913
     - name: Accuracy
       type: accuracy
-      value: 0.2799097065462754
+      value: 0.29232505643340856
     - name: Precision
       type: precision
-      value: 0.7554540842212075
+      value: 0.7679038159958181
     - name: Recall
       type: recall
-      value: 0.6890328551596483
+      value: 0.6797778806108283
 ---
 
 <!-- This model card has been generated automatically according to the information the Trainer had access to. You
@@ -45,12 +45,12 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [SamLowe/roberta-base-go_emotions](https://huggingface.co/SamLowe/roberta-base-go_emotions) on the sem_eval_2018_task_1 dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.3097
-- F1: 0.7207
-- Roc Auc: 0.8127
-- Accuracy: 0.2799
-- Precision: 0.7555
-- Recall: 0.6890
+- Loss: 0.2916
+- F1: 0.7212
+- Roc Auc: 0.8106
+- Accuracy: 0.2923
+- Precision: 0.7679
+- Recall: 0.6798
 
 ## Model description
 
@@ -75,24 +75,20 @@ The following hyperparameters were used during training:
 - seed: 42
 - optimizer: Use OptimizerNames.ADAMW_TORCH with betas=(0.9,0.999) and epsilon=1e-08 and optimizer_args=No additional optimizer arguments
 - lr_scheduler_type: linear
-- num_epochs: 7
+- num_epochs: 3
 
 ### Training results
 
 | Training Loss | Epoch | Step | Validation Loss | F1     | Roc Auc | Accuracy | Precision | Recall |
 |:-------------:|:-----:|:----:|:---------------:|:------:|:-------:|:--------:|:---------:|:------:|
-| 0.3684        | 1.0   | 855  | 0.3003          | 0.7060 | 0.7973  | 0.3070   | 0.7749    | 0.6483 |
-| 0.2776        | 2.0   | 1710 | 0.2930          | 0.7082 | 0.7978  | 0.3025   | 0.7823    | 0.6469 |
-| 0.2441        | 3.0   | 2565 | 0.3019          | 0.7111 | 0.8025  | 0.2968   | 0.7684    | 0.6617 |
-| 0.2205        | 4.0   | 3420 | 0.3008          | 0.7140 | 0.8060  | 0.2698   | 0.7618    | 0.6719 |
-| 0.2002        | 5.0   | 4275 | 0.3058          | 0.7184 | 0.8109  | 0.2709   | 0.7555    | 0.6849 |
-| 0.1844        | 6.0   | 5130 | 0.3097          | 0.7207 | 0.8127  | 0.2799   | 0.7555    | 0.6890 |
-| 0.1692        | 7.0   | 5985 | 0.3110          | 0.7159 | 0.8102  | 0.2709   | 0.7482    | 0.6863 |
+| 0.2744        | 1.0   | 855  | 0.2948          | 0.7061 | 0.7988  | 0.2957   | 0.7668    | 0.6543 |
+| 0.2593        | 2.0   | 1710 | 0.2917          | 0.7144 | 0.8033  | 0.2968   | 0.7781    | 0.6603 |
+| 0.2315        | 3.0   | 2565 | 0.2916          | 0.7212 | 0.8106  | 0.2923   | 0.7679    | 0.6798 |
 
 
 ### Framework versions
 
-- Transformers 4.48.2
-- Pytorch 2.5.1+cu124
-- Datasets 3.2.0
+- Transformers 4.48.3
+- Pytorch 2.6.0+cu124
+- Datasets 3.4.0
 - Tokenizers 0.21.0

config.json

@@ -47,7 +47,7 @@
   "position_embedding_type": "absolute",
   "problem_type": "multi_label_classification",
   "torch_dtype": "float32",
-  "transformers_version": "4.48.2",
+  "transformers_version": "4.48.3",
   "type_vocab_size": 1,
   "use_cache": true,
   "vocab_size": 50265

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:70d77cc6f3efe7c28f0ec234a44868b92ef5b92ea9ed0df2efa784dcd6a12695
+oid sha256:4fe9fb7dc110816e8e342680491efa017661a4ca9ad91d552f48d97605d4a71c
 size 498640508

runs/Mar17_09-31-06_26ab15637bbc/events.out.tfevents.1742203869.26ab15637bbc.678.0

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

runs/Mar17_09-33-05_26ab15637bbc/events.out.tfevents.1742203992.26ab15637bbc.678.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:8ca8c6d2affe743b38a6f87f5c86303569bbc954b77a592c62e7c16c08ba1127
+size 8644

runs/Mar17_09-33-05_26ab15637bbc/events.out.tfevents.1742204616.26ab15637bbc.678.2

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:983da81c2ff558bd84b5a12ff7087cf01ce178dbddc54ebee30a58edd3cb0821
+size 611

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:9359206fc83b24e4eb429234c786235219600cd04a0795c49ea22fe8be1e827d
+oid sha256:9556fe4a8e0e1f3ede68395bf18505d17f3fc680c231e125f2f9cc8b304c82fb
 size 5368