SharonTudi commited on
Commit
4263e87
·
verified ·
1 Parent(s): 0bc3320

End of training

Browse files
README.md CHANGED
@@ -3,11 +3,6 @@ license: apache-2.0
3
  base_model: distilbert-base-cased
4
  tags:
5
  - generated_from_trainer
6
- metrics:
7
- - precision
8
- - recall
9
- - f1
10
- - accuracy
11
  model-index:
12
  - name: DIALOGUE_second_model
13
  results: []
@@ -20,11 +15,16 @@ should probably proofread and complete it, then remove this comment. -->
20
 
21
  This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on the None dataset.
22
  It achieves the following results on the evaluation set:
23
- - Loss: 0.5652
24
- - Precision: 0.8647
25
- - Recall: 0.8553
26
- - F1: 0.8558
27
- - Accuracy: 0.8553
 
 
 
 
 
28
 
29
  ## Model description
30
 
@@ -49,22 +49,7 @@ The following hyperparameters were used during training:
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
52
- - num_epochs: 3
53
-
54
- ### Training results
55
-
56
- | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
57
- |:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
58
- | 1.3504 | 0.31 | 15 | 1.1938 | 0.7324 | 0.5921 | 0.5808 | 0.5921 |
59
- | 1.034 | 0.62 | 30 | 0.8600 | 0.7407 | 0.7105 | 0.7154 | 0.7105 |
60
- | 0.6745 | 0.94 | 45 | 0.6597 | 0.7670 | 0.7632 | 0.7643 | 0.7632 |
61
- | 0.5462 | 1.25 | 60 | 0.6939 | 0.7536 | 0.75 | 0.7432 | 0.75 |
62
- | 0.5193 | 1.56 | 75 | 0.5183 | 0.7839 | 0.7763 | 0.7787 | 0.7763 |
63
- | 0.3645 | 1.88 | 90 | 0.5712 | 0.8512 | 0.8421 | 0.8431 | 0.8421 |
64
- | 0.3355 | 2.19 | 105 | 0.6389 | 0.7836 | 0.7763 | 0.7750 | 0.7763 |
65
- | 0.3992 | 2.5 | 120 | 0.5821 | 0.8180 | 0.8158 | 0.8160 | 0.8158 |
66
- | 0.2971 | 2.81 | 135 | 0.5652 | 0.8647 | 0.8553 | 0.8558 | 0.8553 |
67
-
68
 
69
  ### Framework versions
70
 
 
3
  base_model: distilbert-base-cased
4
  tags:
5
  - generated_from_trainer
 
 
 
 
 
6
  model-index:
7
  - name: DIALOGUE_second_model
8
  results: []
 
15
 
16
  This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on the None dataset.
17
  It achieves the following results on the evaluation set:
18
+ - eval_loss: 1.0869
19
+ - eval_precision: 0.8180
20
+ - eval_recall: 0.8158
21
+ - eval_f1: 0.8160
22
+ - eval_accuracy: 0.8158
23
+ - eval_runtime: 6.0635
24
+ - eval_samples_per_second: 12.534
25
+ - eval_steps_per_second: 1.649
26
+ - epoch: 9.17
27
+ - step: 440
28
 
29
  ## Model description
30
 
 
49
  - seed: 42
50
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
51
  - lr_scheduler_type: linear
52
+ - num_epochs: 15
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
53
 
54
  ### Framework versions
55
 
model.safetensors CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:69a4aae435226db40d6cb3a8d1978218266530aba2f22844ba16caafd8888f0d
3
  size 263150840
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:44bdcde6597ae303b617bee35f310af8eaa3b973406e1c84f1c5e33c5dc46e9f
3
  size 263150840
runs/Jan19_15-28-34_fd24580f6931/events.out.tfevents.1705678117.fd24580f6931.1220.9 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:9632a93b528f09c003c852735010c50c519629537e075a30d88d2a5e81815c72
3
+ size 4841
runs/Jan19_15-34-48_fd24580f6931/events.out.tfevents.1705678491.fd24580f6931.1220.10 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:bfe4bed53a231159706bd55f947ec557d1f17f7b7dc570897e98be8b919e04f9
3
+ size 6271
runs/Jan19_15-35-25_fd24580f6931/events.out.tfevents.1705678528.fd24580f6931.1220.11 ADDED
@@ -0,0 +1,3 @@
 
 
 
 
1
+ version https://git-lfs.github.com/spec/v1
2
+ oid sha256:e42ff3824885d4a00ef34e8a2e1655f57ffd539bc0250428092d816a257faf97
3
+ size 18406
training_args.bin CHANGED
@@ -1,3 +1,3 @@
1
  version https://git-lfs.github.com/spec/v1
2
- oid sha256:3007494585d4fd6ea5069b2c68f4ca03c4e770b6cffa94f2abad7e43302b08b2
3
  size 4664
 
1
  version https://git-lfs.github.com/spec/v1
2
+ oid sha256:56b4b67f6c5e696a88408a4c3d08056ef5660ce1e9333c2009b8bbaf6558042d
3
  size 4664