End of training

README.md

@@ -3,6 +3,11 @@ license: apache-2.0
 base_model: distilbert-base-cased
 tags:
 - generated_from_trainer
+metrics:
+- precision
+- recall
+- f1
+- accuracy
 model-index:
 - name: DIALOGUE_second_model
   results: []
@@ -15,16 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert-base-cased](https://huggingface.co/distilbert-base-cased) on the None dataset.
 It achieves the following results on the evaluation set:
-- eval_loss: 1.0869
-- eval_precision: 0.8180
-- eval_recall: 0.8158
-- eval_f1: 0.8160
-- eval_accuracy: 0.8158
-- eval_runtime: 6.0635
-- eval_samples_per_second: 12.534
-- eval_steps_per_second: 1.649
-- epoch: 9.17
-- step: 440
+- Loss: 0.1121
+- Precision: 0.9762
+- Recall: 0.9737
+- F1: 0.9736
+- Accuracy: 0.9737
 
 ## Model description
 
@@ -43,7 +43,7 @@ More information needed
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 6e-05
+- learning_rate: 1e-05
 - train_batch_size: 8
 - eval_batch_size: 8
 - seed: 42
@@ -51,6 +51,27 @@ The following hyperparameters were used during training:
 - lr_scheduler_type: linear
 - num_epochs: 15
 
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
+| 1.206         | 1.0   | 48   | 0.8496          | 0.8900    | 0.8421 | 0.8322 | 0.8421   |
+| 0.621         | 2.0   | 96   | 0.2971          | 1.0       | 1.0    | 1.0    | 1.0      |
+| 0.2411        | 3.0   | 144  | 0.1237          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0988        | 4.0   | 192  | 0.0829          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0471        | 5.0   | 240  | 0.0926          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0213        | 6.0   | 288  | 0.0924          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0142        | 7.0   | 336  | 0.1157          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0106        | 8.0   | 384  | 0.1067          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0086        | 9.0   | 432  | 0.1032          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.007         | 10.0  | 480  | 0.1188          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0064        | 11.0  | 528  | 0.1134          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0059        | 12.0  | 576  | 0.1122          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0055        | 13.0  | 624  | 0.1109          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0053        | 14.0  | 672  | 0.1125          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+| 0.0052        | 15.0  | 720  | 0.1121          | 0.9762    | 0.9737 | 0.9736 | 0.9737   |
+
+
 ### Framework versions
 
 - Transformers 4.36.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:44bdcde6597ae303b617bee35f310af8eaa3b973406e1c84f1c5e33c5dc46e9f
+oid sha256:23a5643049aca9dca0a9756764cfb7d6f39ce033b1d08cd04a4e1b0dc4c49978
 size 263150840

runs/Jan21_11-29-58_1a3a957c2842/events.out.tfevents.1705836601.1a3a957c2842.8067.1

@@ -0,0 +1,3 @@
+version https://git-lfs.github.com/spec/v1
+oid sha256:ec2a3b5a60f5c3fb99bd1dad0214a4df42c2ff478c03643dc08d81433685ab0f
+size 14261

training_args.bin

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:56b4b67f6c5e696a88408a4c3d08056ef5660ce1e9333c2009b8bbaf6558042d
-size 4664
+oid sha256:d4a7f306b7f3ec24c7f56a4a5a046c71adc46dd964de720e562953f73a7b333a
+size 4728