Commit
•
43eb21b
1
Parent(s):
f196959
update model card README.md
Browse files
README.md
ADDED
@@ -0,0 +1,148 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
1 |
+
---
|
2 |
+
license: mit
|
3 |
+
base_model: roberta-base
|
4 |
+
tags:
|
5 |
+
- generated_from_trainer
|
6 |
+
metrics:
|
7 |
+
- accuracy
|
8 |
+
model-index:
|
9 |
+
- name: roberta-base-emotion-prediction-phr-2
|
10 |
+
results: []
|
11 |
+
---
|
12 |
+
|
13 |
+
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
14 |
+
should probably proofread and complete it, then remove this comment. -->
|
15 |
+
|
16 |
+
# roberta-base-emotion-prediction-phr-2
|
17 |
+
|
18 |
+
This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
|
19 |
+
It achieves the following results on the evaluation set:
|
20 |
+
- Loss: 0.3524
|
21 |
+
- Accuracy: 0.2522
|
22 |
+
- Micro Precision: 0.7220
|
23 |
+
- Micro Recall: 0.6522
|
24 |
+
- Micro F1: 0.6853
|
25 |
+
- Micro Roc Auc: 0.7908
|
26 |
+
|
27 |
+
## Model description
|
28 |
+
|
29 |
+
More information needed
|
30 |
+
|
31 |
+
## Intended uses & limitations
|
32 |
+
|
33 |
+
More information needed
|
34 |
+
|
35 |
+
## Training and evaluation data
|
36 |
+
|
37 |
+
More information needed
|
38 |
+
|
39 |
+
## Training procedure
|
40 |
+
|
41 |
+
### Training hyperparameters
|
42 |
+
|
43 |
+
The following hyperparameters were used during training:
|
44 |
+
- learning_rate: 2e-05
|
45 |
+
- train_batch_size: 8
|
46 |
+
- eval_batch_size: 8
|
47 |
+
- seed: 42
|
48 |
+
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
|
49 |
+
- lr_scheduler_type: linear
|
50 |
+
- num_epochs: 10
|
51 |
+
|
52 |
+
### Training results
|
53 |
+
|
54 |
+
| Training Loss | Epoch | Step | Validation Loss | Accuracy | Micro Precision | Micro Recall | Micro F1 | Micro Roc Auc |
|
55 |
+
|:-------------:|:-----:|:----:|:---------------:|:--------:|:---------------:|:------------:|:--------:|:-------------:|
|
56 |
+
| 0.4952 | 0.12 | 100 | 0.4515 | 0.1574 | 0.5861 | 0.3505 | 0.4386 | 0.6404 |
|
57 |
+
| 0.4152 | 0.23 | 200 | 0.3839 | 0.2041 | 0.7102 | 0.4593 | 0.5578 | 0.7033 |
|
58 |
+
| 0.3878 | 0.35 | 300 | 0.3625 | 0.2341 | 0.7384 | 0.5198 | 0.6101 | 0.7340 |
|
59 |
+
| 0.3764 | 0.47 | 400 | 0.3506 | 0.2412 | 0.7666 | 0.5092 | 0.6119 | 0.7328 |
|
60 |
+
| 0.372 | 0.58 | 500 | 0.3450 | 0.2375 | 0.7686 | 0.5251 | 0.6239 | 0.7403 |
|
61 |
+
| 0.3588 | 0.7 | 600 | 0.3464 | 0.2249 | 0.7804 | 0.4964 | 0.6068 | 0.7286 |
|
62 |
+
| 0.3383 | 0.82 | 700 | 0.3471 | 0.2470 | 0.7503 | 0.5578 | 0.6398 | 0.7528 |
|
63 |
+
| 0.3489 | 0.94 | 800 | 0.3284 | 0.2620 | 0.7702 | 0.5682 | 0.6539 | 0.7603 |
|
64 |
+
| 0.3287 | 1.05 | 900 | 0.3214 | 0.2820 | 0.7707 | 0.5936 | 0.6706 | 0.7720 |
|
65 |
+
| 0.3158 | 1.17 | 1000 | 0.3352 | 0.2657 | 0.7580 | 0.5814 | 0.6580 | 0.7646 |
|
66 |
+
| 0.3247 | 1.29 | 1100 | 0.3219 | 0.2811 | 0.7696 | 0.6031 | 0.6763 | 0.7762 |
|
67 |
+
| 0.3159 | 1.4 | 1200 | 0.3237 | 0.2688 | 0.7479 | 0.6138 | 0.6743 | 0.7778 |
|
68 |
+
| 0.3207 | 1.52 | 1300 | 0.3217 | 0.2461 | 0.7676 | 0.5767 | 0.6586 | 0.7638 |
|
69 |
+
| 0.3087 | 1.64 | 1400 | 0.3253 | 0.2424 | 0.7484 | 0.5883 | 0.6587 | 0.7663 |
|
70 |
+
| 0.3057 | 1.75 | 1500 | 0.3174 | 0.2728 | 0.7587 | 0.6116 | 0.6773 | 0.7785 |
|
71 |
+
| 0.3099 | 1.87 | 1600 | 0.3150 | 0.2774 | 0.7683 | 0.6001 | 0.6738 | 0.7746 |
|
72 |
+
| 0.3006 | 1.99 | 1700 | 0.3176 | 0.2633 | 0.7636 | 0.5881 | 0.6645 | 0.7685 |
|
73 |
+
| 0.285 | 2.11 | 1800 | 0.3177 | 0.2722 | 0.7363 | 0.6484 | 0.6896 | 0.7915 |
|
74 |
+
| 0.2886 | 2.22 | 1900 | 0.3156 | 0.2768 | 0.7734 | 0.5935 | 0.6716 | 0.7723 |
|
75 |
+
| 0.2785 | 2.34 | 2000 | 0.3101 | 0.2808 | 0.7692 | 0.6151 | 0.6836 | 0.7816 |
|
76 |
+
| 0.2801 | 2.46 | 2100 | 0.3121 | 0.2728 | 0.7739 | 0.5956 | 0.6732 | 0.7734 |
|
77 |
+
| 0.2876 | 2.57 | 2200 | 0.3166 | 0.2777 | 0.7577 | 0.6157 | 0.6794 | 0.7802 |
|
78 |
+
| 0.2769 | 2.69 | 2300 | 0.3143 | 0.2881 | 0.7691 | 0.6124 | 0.6819 | 0.7803 |
|
79 |
+
| 0.2755 | 2.81 | 2400 | 0.3133 | 0.2792 | 0.7577 | 0.6263 | 0.6857 | 0.7850 |
|
80 |
+
| 0.2815 | 2.92 | 2500 | 0.3197 | 0.2716 | 0.7406 | 0.6466 | 0.6904 | 0.7914 |
|
81 |
+
| 0.2671 | 3.04 | 2600 | 0.3133 | 0.2857 | 0.7549 | 0.6438 | 0.6949 | 0.7925 |
|
82 |
+
| 0.2431 | 3.16 | 2700 | 0.3225 | 0.2722 | 0.7515 | 0.6320 | 0.6866 | 0.7866 |
|
83 |
+
| 0.2512 | 3.27 | 2800 | 0.3221 | 0.2743 | 0.7616 | 0.6106 | 0.6778 | 0.7784 |
|
84 |
+
| 0.2574 | 3.39 | 2900 | 0.3191 | 0.2737 | 0.7561 | 0.6214 | 0.6822 | 0.7825 |
|
85 |
+
| 0.2527 | 3.51 | 3000 | 0.3207 | 0.2666 | 0.7443 | 0.6315 | 0.6833 | 0.7852 |
|
86 |
+
| 0.2615 | 3.63 | 3100 | 0.3170 | 0.2670 | 0.7443 | 0.6471 | 0.6923 | 0.7923 |
|
87 |
+
| 0.2583 | 3.74 | 3200 | 0.3122 | 0.2685 | 0.7729 | 0.6068 | 0.6799 | 0.7783 |
|
88 |
+
| 0.2543 | 3.86 | 3300 | 0.3175 | 0.2709 | 0.7492 | 0.6432 | 0.6921 | 0.7913 |
|
89 |
+
| 0.2546 | 3.98 | 3400 | 0.3164 | 0.2752 | 0.7661 | 0.6186 | 0.6845 | 0.7828 |
|
90 |
+
| 0.2274 | 4.09 | 3500 | 0.3172 | 0.2759 | 0.7437 | 0.6426 | 0.6895 | 0.7902 |
|
91 |
+
| 0.2328 | 4.21 | 3600 | 0.3214 | 0.2737 | 0.7548 | 0.6297 | 0.6866 | 0.7861 |
|
92 |
+
| 0.2354 | 4.33 | 3700 | 0.3192 | 0.2792 | 0.7546 | 0.6310 | 0.6872 | 0.7866 |
|
93 |
+
| 0.2238 | 4.44 | 3800 | 0.3199 | 0.2709 | 0.7453 | 0.6444 | 0.6912 | 0.7912 |
|
94 |
+
| 0.2376 | 4.56 | 3900 | 0.3176 | 0.2734 | 0.7599 | 0.6247 | 0.6857 | 0.7846 |
|
95 |
+
| 0.2344 | 4.68 | 4000 | 0.3189 | 0.2639 | 0.7437 | 0.6390 | 0.6874 | 0.7885 |
|
96 |
+
| 0.2222 | 4.8 | 4100 | 0.3222 | 0.2636 | 0.7436 | 0.6409 | 0.6884 | 0.7894 |
|
97 |
+
| 0.232 | 4.91 | 4200 | 0.3227 | 0.2725 | 0.7472 | 0.6426 | 0.6910 | 0.7907 |
|
98 |
+
| 0.2367 | 5.03 | 4300 | 0.3243 | 0.2670 | 0.7463 | 0.6339 | 0.6855 | 0.7866 |
|
99 |
+
| 0.2154 | 5.15 | 4400 | 0.3257 | 0.2593 | 0.7366 | 0.6513 | 0.6913 | 0.7929 |
|
100 |
+
| 0.2089 | 5.26 | 4500 | 0.3261 | 0.2700 | 0.7416 | 0.6453 | 0.6901 | 0.7910 |
|
101 |
+
| 0.2081 | 5.38 | 4600 | 0.3269 | 0.2731 | 0.7602 | 0.6133 | 0.6789 | 0.7794 |
|
102 |
+
| 0.2116 | 5.5 | 4700 | 0.3308 | 0.2593 | 0.7229 | 0.6687 | 0.6947 | 0.7983 |
|
103 |
+
| 0.2128 | 5.61 | 4800 | 0.3263 | 0.2660 | 0.7422 | 0.6432 | 0.6891 | 0.7902 |
|
104 |
+
| 0.2059 | 5.73 | 4900 | 0.3295 | 0.2728 | 0.7356 | 0.6550 | 0.6929 | 0.7944 |
|
105 |
+
| 0.2103 | 5.85 | 5000 | 0.3301 | 0.2814 | 0.7442 | 0.6510 | 0.6945 | 0.7940 |
|
106 |
+
| 0.2151 | 5.96 | 5100 | 0.3300 | 0.2541 | 0.7221 | 0.6598 | 0.6896 | 0.7942 |
|
107 |
+
| 0.1954 | 6.08 | 5200 | 0.3325 | 0.2765 | 0.7476 | 0.6381 | 0.6885 | 0.7887 |
|
108 |
+
| 0.2028 | 6.2 | 5300 | 0.3316 | 0.2559 | 0.7364 | 0.6400 | 0.6848 | 0.7878 |
|
109 |
+
| 0.1911 | 6.32 | 5400 | 0.3332 | 0.2553 | 0.7370 | 0.6386 | 0.6843 | 0.7873 |
|
110 |
+
| 0.2015 | 6.43 | 5500 | 0.3349 | 0.2645 | 0.7308 | 0.6538 | 0.6902 | 0.7931 |
|
111 |
+
| 0.1901 | 6.55 | 5600 | 0.3389 | 0.2587 | 0.7197 | 0.6682 | 0.6930 | 0.7975 |
|
112 |
+
| 0.197 | 6.67 | 5700 | 0.3349 | 0.2728 | 0.7400 | 0.6424 | 0.6878 | 0.7895 |
|
113 |
+
| 0.1907 | 6.78 | 5800 | 0.3354 | 0.2627 | 0.7454 | 0.6349 | 0.6857 | 0.7870 |
|
114 |
+
| 0.1853 | 6.9 | 5900 | 0.3420 | 0.2657 | 0.7356 | 0.6513 | 0.6909 | 0.7927 |
|
115 |
+
| 0.1841 | 7.02 | 6000 | 0.3399 | 0.2584 | 0.7308 | 0.6554 | 0.6910 | 0.7937 |
|
116 |
+
| 0.1739 | 7.13 | 6100 | 0.3409 | 0.2620 | 0.7364 | 0.6446 | 0.6874 | 0.7898 |
|
117 |
+
| 0.1768 | 7.25 | 6200 | 0.3417 | 0.2593 | 0.7314 | 0.6474 | 0.6868 | 0.7902 |
|
118 |
+
| 0.1762 | 7.37 | 6300 | 0.3384 | 0.2654 | 0.7398 | 0.6373 | 0.6847 | 0.7871 |
|
119 |
+
| 0.177 | 7.49 | 6400 | 0.3448 | 0.2541 | 0.7237 | 0.6547 | 0.6875 | 0.7922 |
|
120 |
+
| 0.1787 | 7.6 | 6500 | 0.3422 | 0.2513 | 0.7317 | 0.6425 | 0.6842 | 0.7881 |
|
121 |
+
| 0.1793 | 7.72 | 6600 | 0.3452 | 0.2611 | 0.7231 | 0.6582 | 0.6891 | 0.7936 |
|
122 |
+
| 0.1772 | 7.84 | 6700 | 0.3470 | 0.2587 | 0.7193 | 0.6618 | 0.6894 | 0.7946 |
|
123 |
+
| 0.1799 | 7.95 | 6800 | 0.3459 | 0.2547 | 0.7238 | 0.6494 | 0.6846 | 0.7898 |
|
124 |
+
| 0.1726 | 8.07 | 6900 | 0.3477 | 0.2507 | 0.7259 | 0.6419 | 0.6813 | 0.7869 |
|
125 |
+
| 0.1672 | 8.19 | 7000 | 0.3489 | 0.2492 | 0.7215 | 0.6499 | 0.6838 | 0.7897 |
|
126 |
+
| 0.1664 | 8.3 | 7100 | 0.3474 | 0.2498 | 0.7197 | 0.6491 | 0.6826 | 0.7890 |
|
127 |
+
| 0.1712 | 8.42 | 7200 | 0.3477 | 0.2516 | 0.7309 | 0.6404 | 0.6827 | 0.7870 |
|
128 |
+
| 0.166 | 8.54 | 7300 | 0.3487 | 0.2553 | 0.7209 | 0.6547 | 0.6862 | 0.7917 |
|
129 |
+
| 0.1706 | 8.65 | 7400 | 0.3487 | 0.2538 | 0.7239 | 0.6518 | 0.6860 | 0.7909 |
|
130 |
+
| 0.1674 | 8.77 | 7500 | 0.3506 | 0.2538 | 0.7216 | 0.6541 | 0.6862 | 0.7916 |
|
131 |
+
| 0.1655 | 8.89 | 7600 | 0.3476 | 0.2553 | 0.7283 | 0.6465 | 0.6849 | 0.7893 |
|
132 |
+
| 0.1609 | 9.01 | 7700 | 0.3498 | 0.2495 | 0.7273 | 0.6443 | 0.6833 | 0.7882 |
|
133 |
+
| 0.1647 | 9.12 | 7800 | 0.3507 | 0.2522 | 0.7255 | 0.6423 | 0.6814 | 0.7870 |
|
134 |
+
| 0.1531 | 9.24 | 7900 | 0.3503 | 0.2522 | 0.7292 | 0.6426 | 0.6832 | 0.7878 |
|
135 |
+
| 0.1577 | 9.36 | 8000 | 0.3524 | 0.2528 | 0.7212 | 0.6569 | 0.6875 | 0.7927 |
|
136 |
+
| 0.1592 | 9.47 | 8100 | 0.3517 | 0.2519 | 0.7186 | 0.6536 | 0.6845 | 0.7908 |
|
137 |
+
| 0.1615 | 9.59 | 8200 | 0.3514 | 0.2510 | 0.7183 | 0.6529 | 0.6841 | 0.7905 |
|
138 |
+
| 0.1529 | 9.71 | 8300 | 0.3515 | 0.2516 | 0.7221 | 0.6489 | 0.6835 | 0.7893 |
|
139 |
+
| 0.1607 | 9.82 | 8400 | 0.3520 | 0.2528 | 0.7212 | 0.6499 | 0.6837 | 0.7896 |
|
140 |
+
| 0.1506 | 9.94 | 8500 | 0.3524 | 0.2522 | 0.7220 | 0.6522 | 0.6853 | 0.7908 |
|
141 |
+
|
142 |
+
|
143 |
+
### Framework versions
|
144 |
+
|
145 |
+
- Transformers 4.31.0
|
146 |
+
- Pytorch 2.1.0+cu121
|
147 |
+
- Datasets 2.14.5
|
148 |
+
- Tokenizers 0.13.3
|