tasinhoque
commited on
Commit
•
8ae90f8
1
Parent(s):
77c9fa3
Update README.md
Browse files
README.md
CHANGED
@@ -27,13 +27,15 @@ model-index:
|
|
27 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
28 |
should probably proofread and complete it, then remove this comment. -->
|
29 |
|
30 |
-
#
|
31 |
|
32 |
This model is a fine-tuned version of [roberta-large](https://huggingface.co/roberta-large) on the [go_emotions](https://huggingface.co/datasets/go_emotions) dataset.
|
33 |
|
34 |
## Model description
|
35 |
|
36 |
-
|
|
|
|
|
37 |
|
38 |
## Intended uses & limitations
|
39 |
|
|
|
27 |
<!-- This model card has been generated automatically according to the information the Trainer had access to. You
|
28 |
should probably proofread and complete it, then remove this comment. -->
|
29 |
|
30 |
+
# Text Classification GoEmotions
|
31 |
|
32 |
This model is a fine-tuned version of [roberta-large](https://huggingface.co/roberta-large) on the [go_emotions](https://huggingface.co/datasets/go_emotions) dataset.
|
33 |
|
34 |
## Model description
|
35 |
|
36 |
+
At first, 4 epochs of training with a learning rate of 5e-5 was performed on the `roberta-large` model.
|
37 |
+
After that, the weights were loaded in a new environment and another epoch of training was done (this time with a learning rate of 2e-5).
|
38 |
+
As the performance decreased in the fifth epoch, further training was discontinued.
|
39 |
|
40 |
## Intended uses & limitations
|
41 |
|