arnabdhar commited on
Commit
a08502b
1 Parent(s): 4da18fe

End of training

Browse files
README.md ADDED
@@ -0,0 +1,75 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: apache-2.0
3
+ base_model: distilbert-base-multilingual-cased
4
+ tags:
5
+ - pytorch
6
+ - amazon-rating
7
+ - DistilBERTForSequenceClassification
8
+ - generated_from_trainer
9
+ metrics:
10
+ - accuracy
11
+ - matthews_correlation
12
+ model-index:
13
+ - name: distilbert-base-amazon-multi
14
+ results: []
15
+ ---
16
+
17
+ <!-- This model card has been generated automatically according to the information the Trainer had access to. You
18
+ should probably proofread and complete it, then remove this comment. -->
19
+
20
+ # distilbert-base-amazon-multi
21
+
22
+ This model is a fine-tuned version of [distilbert-base-multilingual-cased](https://huggingface.co/distilbert-base-multilingual-cased) on the mteb/amazon_reviews_multi dataset.
23
+ It achieves the following results on the evaluation set:
24
+ - Loss: 0.9292
25
+ - Accuracy: 0.6055
26
+ - Matthews Correlation: 0.5072
27
+
28
+ ## Model description
29
+
30
+ More information needed
31
+
32
+ ## Intended uses & limitations
33
+
34
+ More information needed
35
+
36
+ ## Training and evaluation data
37
+
38
+ More information needed
39
+
40
+ ## Training procedure
41
+
42
+ ### Training hyperparameters
43
+
44
+ The following hyperparameters were used during training:
45
+ - learning_rate: 5e-05
46
+ - train_batch_size: 32
47
+ - eval_batch_size: 320
48
+ - seed: 42
49
+ - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
+ - lr_scheduler_type: cosine
51
+ - lr_scheduler_warmup_ratio: 0.1
52
+ - training_steps: 100000
53
+
54
+ ### Training results
55
+
56
+ | Training Loss | Epoch | Step | Validation Loss | Accuracy | Matthews Correlation |
57
+ |:-------------:|:-----:|:------:|:---------------:|:--------:|:--------------------:|
58
+ | 1.0008 | 0.26 | 10000 | 1.0027 | 0.5616 | 0.4520 |
59
+ | 0.9545 | 0.51 | 20000 | 0.9705 | 0.5810 | 0.4788 |
60
+ | 0.9216 | 0.77 | 30000 | 0.9415 | 0.5883 | 0.4868 |
61
+ | 0.8765 | 1.03 | 40000 | 0.9495 | 0.5891 | 0.4871 |
62
+ | 0.8837 | 1.28 | 50000 | 0.9254 | 0.5992 | 0.4997 |
63
+ | 0.8753 | 1.54 | 60000 | 0.9199 | 0.6014 | 0.5029 |
64
+ | 0.8572 | 1.8 | 70000 | 0.9108 | 0.6090 | 0.5117 |
65
+ | 0.7851 | 2.05 | 80000 | 0.9276 | 0.6052 | 0.5066 |
66
+ | 0.7918 | 2.31 | 90000 | 0.9292 | 0.6055 | 0.5072 |
67
+ | 0.793 | 2.57 | 100000 | 0.9288 | 0.6064 | 0.5084 |
68
+
69
+
70
+ ### Framework versions
71
+
72
+ - Transformers 4.35.2
73
+ - Pytorch 2.1.0+cu118
74
+ - Datasets 2.15.0
75
+ - Tokenizers 0.15.0
evaluate_results.json ADDED
@@ -0,0 +1,9 @@
 
 
 
 
 
 
 
 
 
 
1
+ {
2
+ "epoch": 2.57,
3
+ "eval_accuracy": 0.6054761904761905,
4
+ "eval_loss": 0.9292314648628235,
5
+ "eval_matthews_correlation": 0.5072025915463576,
6
+ "eval_runtime": 63.8498,
7
+ "eval_samples_per_second": 197.338,
8
+ "eval_steps_per_second": 0.626
9
+ }
runs/Dec02_02-38-05_4888c9987e8e/events.out.tfevents.1701484697.4888c9987e8e.398.0 CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:13df6100d76df8f5ffd70308359d3baa059720b8937c8d822ad88ddd74eed012
3
- size 36661
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:ac972e5fc7a347a362047525e95ba719e6ca88b209f324cebb05f472bc0daf23
3
+ size 40615
runs/Dec02_02-38-05_4888c9987e8e/events.out.tfevents.1701531798.4888c9987e8e.398.1 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:3402c5e069e9b6a390cae70adb10e53a3ae000be1709d62ac86e7bf9be5eb5ed
3
+ size 482