Hugging Face's logo

lewtun
/
roberta-base-bne-finetuned-amazon_reviews_multi

Text Classification
Transformers
PyTorch
TensorBoard
roberta
generated_from_trainer
Inference Endpoints
Model card Files Metrics Community
2
lewtun HF staff commited on
Commit
48a974e
•
1 Parent(s): ff4d035

add model

Browse files
Files changed (7) hide show
  1. README.md +5 -5
  2. config.json +1 -9
  3. pytorch_model.bin +1 -1
  4. runs/Aug23_16-43-12_7843dcbb7c4b/1629737151.2558837/events.out.tfevents.1629737151.7843dcbb7c4b.76.1 +3 -0
  5. runs/Aug23_16-43-12_7843dcbb7c4b/events.out.tfevents.1629737151.7843dcbb7c4b.76.0 +3 -0
  6. tokenizer.json +0 -0
  7. training_args.bin +1 -1
README.md CHANGED
@@ -29,7 +29,7 @@ should probably proofread and complete it, then remove this comment. -->
29
 
30
  This model is a fine-tuned version of [BSC-TeMU/roberta-base-bne](https://huggingface.co/BSC-TeMU/roberta-base-bne) on the amazon_reviews_multi dataset.
31
  It achieves the following results on the evaluation set:
32
- - Loss: 0.2005
33
  - Accuracy: 0.9307
34
 
35
  ## Model description
@@ -50,8 +50,8 @@ More information needed
50
 
51
  The following hyperparameters were used during training:
52
  - learning_rate: 2e-05
53
- - train_batch_size: 32
54
- - eval_batch_size: 32
55
  - seed: 42
56
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
57
  - lr_scheduler_type: linear
@@ -61,8 +61,8 @@ The following hyperparameters were used during training:
61
 
62
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
63
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
64
- | 0.1776 | 1.0 | 625 | 0.1761 | 0.9295 |
65
- | 0.0924 | 2.0 | 1250 | 0.2005 | 0.9307 |
66
 
67
 
68
  ### Framework versions
 
29
 
30
  This model is a fine-tuned version of [BSC-TeMU/roberta-base-bne](https://huggingface.co/BSC-TeMU/roberta-base-bne) on the amazon_reviews_multi dataset.
31
  It achieves the following results on the evaluation set:
32
+ - Loss: 0.2306
33
  - Accuracy: 0.9307
34
 
35
  ## Model description
 
50
 
51
  The following hyperparameters were used during training:
52
  - learning_rate: 2e-05
53
+ - train_batch_size: 16
54
+ - eval_batch_size: 16
55
  - seed: 42
56
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
57
  - lr_scheduler_type: linear
 
61
 
62
  | Training Loss | Epoch | Step | Validation Loss | Accuracy |
63
  |:-------------:|:-----:|:----:|:---------------:|:--------:|
64
+ | 0.1978 | 1.0 | 1250 | 0.1750 | 0.9325 |
65
+ | 0.0951 | 2.0 | 2500 | 0.2306 | 0.9307 |
66
 
67
 
68
  ### Framework versions
config.json CHANGED
@@ -1,5 +1,5 @@
1
  {
2
- "_name_or_path": "lewtun/roberta-base-bne-finetuned-amazon_reviews_multi",
3
  "architectures": [
4
  "RobertaForSequenceClassification"
5
  ],
@@ -10,16 +10,8 @@
10
  "hidden_act": "gelu",
11
  "hidden_dropout_prob": 0.0,
12
  "hidden_size": 768,
13
- "id2label": {
14
- "0": "NEGATIVO",
15
- "1": "POSITIVO"
16
- },
17
  "initializer_range": 0.02,
18
  "intermediate_size": 3072,
19
- "label2id": {
20
- "NEGATIVO": 0,
21
- "POSITIVO": 1
22
- },
23
  "layer_norm_eps": 1e-05,
24
  "max_position_embeddings": 514,
25
  "model_type": "roberta",
 
1
  {
2
+ "_name_or_path": "BSC-TeMU/roberta-base-bne",
3
  "architectures": [
4
  "RobertaForSequenceClassification"
5
  ],
 
10
  "hidden_act": "gelu",
11
  "hidden_dropout_prob": 0.0,
12
  "hidden_size": 768,
 
 
 
 
13
  "initializer_range": 0.02,
14
  "intermediate_size": 3072,
 
 
 
 
15
  "layer_norm_eps": 1e-05,
16
  "max_position_embeddings": 514,
17
  "model_type": "roberta",
pytorch_model.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:e901750ea4e15da1584b754d7c53e92422734ac5344b6d4a2d4c1dc03b7bb89b
3
  size 498664877
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:1a1fb568060783cec4f9fd55fc41a2cb7c9affb6c708a80ac9728613638dff4f
3
  size 498664877
runs/Aug23_16-43-12_7843dcbb7c4b/1629737151.2558837/events.out.tfevents.1629737151.7843dcbb7c4b.76.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:fc44bcda71c27950a2ad2e36dd62aa3db119ddea6edb11fa0e5877cbb0664a85
3
+ size 4212
runs/Aug23_16-43-12_7843dcbb7c4b/events.out.tfevents.1629737151.7843dcbb7c4b.76.0 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:2b495e10313d6e0d13dc4cb5f81ee4321532438ce0f572fa034d406e95f04929
3
+ size 5252
tokenizer.json CHANGED
The diff for this file is too large to render. See raw diff
 
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:21cf3968778a76803fb46cc632bbd700d4a608ce95825181a1bda7a1bdf20102
3
  size 2671
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44dc4e25b7af53909b1c55a0e4cbaaf7a5aa88b50df783c3021b2d9e2c968e2f
3
  size 2671