Hugging Face's logo

slickdata
/
finetuned-Sentiment-classfication-ROBERTA-model

Text Classification
Transformers
PyTorch
TensorBoard
roberta
Generated from Trainer
Inference Endpoints
Model card Files Metrics Community
slickdata commited on
Commit
24e64dc
•
1 Parent(s): e437b06

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -5
README.md CHANGED
@@ -1,5 +1,6 @@
1
  ---
2
  license: mit
 
3
  tags:
4
  - generated_from_trainer
5
  model-index:
@@ -14,8 +15,8 @@ should probably proofread and complete it, then remove this comment. -->
14
 
15
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
16
  It achieves the following results on the evaluation set:
17
- - Loss: 0.6126
18
- - Rmse: 0.6362
19
 
20
  ## Model description
21
 
@@ -43,18 +44,22 @@ The following hyperparameters were used during training:
43
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
44
  - lr_scheduler_type: linear
45
  - lr_scheduler_warmup_steps: 500
46
- - num_epochs: 7
47
 
48
  ### Training results
49
 
50
  | Training Loss | Epoch | Step | Validation Loss | Rmse |
51
  |:-------------:|:-----:|:----:|:---------------:|:------:|
52
- | 0.6393 | 4.0 | 500 | 0.6126 | 0.6362 |
 
 
 
 
53
 
54
 
55
  ### Framework versions
56
 
57
- - Transformers 4.30.2
58
  - Pytorch 2.0.1+cu118
59
  - Datasets 2.13.1
60
  - Tokenizers 0.13.3
 
1
  ---
2
  license: mit
3
+ base_model: roberta-base
4
  tags:
5
  - generated_from_trainer
6
  model-index:
 
15
 
16
  This model is a fine-tuned version of [roberta-base](https://huggingface.co/roberta-base) on an unknown dataset.
17
  It achieves the following results on the evaluation set:
18
+ - Loss: 0.2222
19
+ - Rmse: 0.2936
20
 
21
  ## Model description
22
 
 
44
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
45
  - lr_scheduler_type: linear
46
  - lr_scheduler_warmup_steps: 500
47
+ - num_epochs: 16
48
 
49
  ### Training results
50
 
51
  | Training Loss | Epoch | Step | Validation Loss | Rmse |
52
  |:-------------:|:-----:|:----:|:---------------:|:------:|
53
+ | 0.6684 | 2.72 | 500 | 0.3931 | 0.4892 |
54
+ | 0.1963 | 5.43 | 1000 | 0.2222 | 0.2936 |
55
+ | 0.0755 | 8.15 | 1500 | 0.2479 | 0.2757 |
56
+ | 0.0413 | 10.86 | 2000 | 0.3233 | 0.2794 |
57
+ | 0.0213 | 13.58 | 2500 | 0.3590 | 0.2689 |
58
 
59
 
60
  ### Framework versions
61
 
62
+ - Transformers 4.31.0
63
  - Pytorch 2.0.1+cu118
64
  - Datasets 2.13.1
65
  - Tokenizers 0.13.3