tasinhoque commited on
Commit
8ae90f8
1 Parent(s): 77c9fa3

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +4 -2
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
- # 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
- More information needed
 
 
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