End of training

README.md

@@ -1,13 +1,8 @@
 ---
 license: apache-2.0
-base_model: bert-large-uncased
+base_model: distilbert-base-uncased
 tags:
 - generated_from_trainer
-metrics:
-- precision
-- recall
-- f1
-- accuracy
 model-index:
 - name: my_model
   results: []
@@ -18,13 +13,9 @@ should probably proofread and complete it, then remove this comment. -->
 
 # my_model
 
-This model is a fine-tuned version of [bert-large-uncased](https://huggingface.co/bert-large-uncased) on the None dataset.
+This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on the None dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.0006
-- Precision: 0.7888
-- Recall: 0.8082
-- F1: 0.7984
-- Accuracy: 0.9998
+- Loss: 0.0015
 
 ## Model description
 
@@ -44,24 +35,25 @@ More information needed
 
 The following hyperparameters were used during training:
 - learning_rate: 2e-05
-- train_batch_size: 4
-- eval_batch_size: 4
+- train_batch_size: 16
+- eval_batch_size: 16
 - seed: 42
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 2
+- num_epochs: 3
 
 ### Training results
 
-| Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1     | Accuracy |
-|:-------------:|:-----:|:----:|:---------------:|:---------:|:------:|:------:|:--------:|
-| 0.0017        | 1.0   | 1362 | 0.0010          | 0.5958    | 0.6980 | 0.6429 | 0.9997   |
-| 0.0006        | 2.0   | 2724 | 0.0006          | 0.7888    | 0.8082 | 0.7984 | 0.9998   |
+| Training Loss | Epoch | Step | Validation Loss |
+|:-------------:|:-----:|:----:|:---------------:|
+| No log        | 1.0   | 383  | 0.0025          |
+| 0.0304        | 2.0   | 766  | 0.0016          |
+| 0.0012        | 3.0   | 1149 | 0.0015          |
 
 
 ### Framework versions
 
-- Transformers 4.35.2
-- Pytorch 2.1.0+cu121
-- Datasets 2.16.1
-- Tokenizers 0.15.1
+- Transformers 4.38.2
+- Pytorch 2.2.1+cu121
+- Datasets 2.18.0
+- Tokenizers 0.15.2

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:b22da7496a9b097570953434d79dc3bdfc7f9f63e9753964809861d5fb7bf4f1
+oid sha256:ccf3e7277dd35a8177fe42801fe59a6985a66a8078e027c93002d07b11b3ff7c
 size 265503852

runs/Mar29_03-29-18_341bed32227e/events.out.tfevents.1711682960.341bed32227e.152.0

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:dbf03ceb8a2466b4b8a7a6cbd348a2f257c2dbd33439a129ef9d4311a2ce2905
-size 5338
+oid sha256:59761d8c95d39f4ca6b1da61858ba19cf1cb49bd4353ad5be65e20522f197ba1
+size 6656