End of training

README.md

@@ -20,11 +20,11 @@ should probably proofread and complete it, then remove this comment. -->
 
 This model is a fine-tuned version of [distilbert/distilbert-base-uncased](https://huggingface.co/distilbert/distilbert-base-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5940
-- Accuracy: 0.6330
-- F1: 0.5455
-- Precision: 0.5052
-- Recall: 0.6330
+- Loss: 0.6515
+- Accuracy: 0.6429
+- F1: 0.5562
+- Precision: 0.5129
+- Recall: 0.6429
 
 ## Model description
 
@@ -44,15 +44,19 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 5e-05
-- train_batch_size: 64
-- eval_batch_size: 64
+- train_batch_size: 16
+- eval_batch_size: 32
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 3
+- num_epochs: 5
 
 ### Training results
 
+| Training Loss | Epoch | Step | Validation Loss | Accuracy | F1     | Precision | Recall |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|:------:|:---------:|:------:|
+| 0.6062        | 2.49  | 500  | 0.6515          | 0.6429   | 0.5562 | 0.5129    | 0.6429 |
+| 0.42          | 4.98  | 1000 | 0.7364          | 0.6454   | 0.5587 | 0.5154    | 0.6454 |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c0af14e2e3e68ca9968529fa76b351569650f45227058ada5e16d150120a82f9
+oid sha256:4aeda606cbc083a7b8c15b9d7d98f6585eaf906c6157bbfa4f4cccccecec01e5
 size 267838720

runs/May19_13-28-10_4e279a7604df/events.out.tfevents.1716125433.4e279a7604df.34.7

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