End of training

README.md

@@ -1,6 +1,6 @@
 ---
-license: cc-by-nc-4.0
-base_model: AIMH/mental-bert-base-cased
+license: apache-2.0
+base_model: google-bert/bert-base-uncased
 tags:
 - generated_from_trainer
 metrics:
@@ -15,10 +15,10 @@ should probably proofread and complete it, then remove this comment. -->
 
 # BERTForDetectingDepression-Twitter2015
 
-This model is a fine-tuned version of [AIMH/mental-bert-base-cased](https://huggingface.co/AIMH/mental-bert-base-cased) on [this dataset](https://www.kaggle.com/datasets/infamouscoder/mental-health-social-media).
+This model is a fine-tuned version of [google-bert/bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased) on an unknown dataset.
 It achieves the following results on the evaluation set:
-- Loss: 0.5232
-- Accuracy: 0.9276
+- Loss: 0.2785
+- Accuracy: 0.9214
 
 ## Model description
 
@@ -30,33 +30,27 @@ More information needed
 
 ## Training and evaluation data
 
-Eval Precision: 0.9206730769230769
-
-Eval Recall: 0.9364303178484108
-
-Eval Accuracy: 0.9275629220380601
+More information needed
 
 ## Training procedure
 
 ### Training hyperparameters
 
 The following hyperparameters were used during training:
-- learning_rate: 1.8919952478487808e-05
-- train_batch_size: 4
+- learning_rate: 7.022911988641232e-05
+- train_batch_size: 16
 - eval_batch_size: 16
-- seed: 36
+- seed: 15
 - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
 - lr_scheduler_type: linear
-- num_epochs: 4
+- num_epochs: 2
 
 ### Training results
 
-| Training Loss | Epoch | Step  | Validation Loss | Accuracy |
-|:-------------:|:-----:|:-----:|:---------------:|:--------:|
-| 0.3347        | 1.0   | 3665  | 0.3384          | 0.9061   |
-| 0.152         | 2.0   | 7330  | 0.4665          | 0.9227   |
-| 0.0444        | 3.0   | 10995 | 0.5232          | 0.9276   |
-| 0.0057        | 4.0   | 14660 | 0.5890          | 0.9269   |
+| Training Loss | Epoch | Step | Validation Loss | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:--------:|
+| 0.3365        | 1.0   | 917  | 0.2187          | 0.9128   |
+| 0.1213        | 2.0   | 1834 | 0.2785          | 0.9214   |
 
 
 ### Framework versions

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:e18b3b55391290fa102826a1e71aba3750b61eafc8d5a853f755688f29389482
+oid sha256:6997d4b0825355d192e419b5fdd8849af76851f2ff296cf80949f1b89da04c8a
 size 437958648

runs/Aug02_03-09-56_8953887c6028/events.out.tfevents.1722569284.8953887c6028.1235.11

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