End of training

README.md

@@ -0,0 +1,75 @@
+---
+license: apache-2.0
+base_model: google-t5/t5-base
+tags:
+- generated_from_trainer
+metrics:
+- rouge
+- precision
+- recall
+- f1
+- accuracy
+model-index:
+- name: t5-base-finetuned-depression
+  results: []
+---
+
+<!-- This model card has been generated automatically according to the information the Trainer had access to. You
+should probably proofread and complete it, then remove this comment. -->
+
+# t5-base-finetuned-depression
+
+This model is a fine-tuned version of [google-t5/t5-base](https://huggingface.co/google-t5/t5-base) on an unknown dataset.
+It achieves the following results on the evaluation set:
+- Loss: 0.3483
+- Rouge1: 67.6667
+- Rouge2: 21.8889
+- Rougel: 67.6667
+- Rougelsum: 67.6667
+- Gen Len: 2.2944
+- Precision: 0.6767
+- Recall: 0.6767
+- F1: 0.6767
+- Accuracy: 0.6767
+
+## Model description
+
+More information needed
+
+## Intended uses & limitations
+
+More information needed
+
+## Training and evaluation data
+
+More information needed
+
+## Training procedure
+
+### Training hyperparameters
+
+The following hyperparameters were used during training:
+- learning_rate: 2e-05
+- train_batch_size: 3
+- eval_batch_size: 3
+- seed: 42
+- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
+- lr_scheduler_type: linear
+- num_epochs: 3
+- mixed_precision_training: Native AMP
+
+### Training results
+
+| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2  | Rougel  | Rougelsum | Gen Len | Precision | Recall | F1     | Accuracy |
+|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|:---------:|:------:|:------:|:--------:|
+| 0.3908        | 1.0   | 1199 | 0.3346          | 67.1111 | 20.0    | 67.3333 | 67.2222   | 2.2578  | 0.6722    | 0.6722 | 0.6722 | 0.6722   |
+| 0.3601        | 2.0   | 2398 | 0.3582          | 67.4444 | 18.5556 | 67.5556 | 67.6667   | 2.2256  | 0.6756    | 0.6756 | 0.6756 | 0.6756   |
+| 0.3289        | 3.0   | 3597 | 0.3483          | 67.6667 | 21.8889 | 67.6667 | 67.6667   | 2.2944  | 0.6767    | 0.6767 | 0.6767 | 0.6767   |
+
+
+### Framework versions
+
+- Transformers 4.41.1
+- Pytorch 2.3.0+cu121
+- Datasets 2.19.1
+- Tokenizers 0.19.1

generation_config.json

@@ -0,0 +1,6 @@
+{
+  "decoder_start_token_id": 0,
+  "eos_token_id": 1,
+  "pad_token_id": 0,
+  "transformers_version": "4.41.1"
+}

model.safetensors

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:37a365c5f211c4495dc7c99b818b290f5a5b9fc2d4d832fb2cbe462348591e53
+oid sha256:1842317db081b622577105e96e1ed0896c90f546adb676e31b30ea982c641941
 size 891644712

runs/Jun02_19-22-23_085015c5a530/events.out.tfevents.1717358459.085015c5a530.298.10

@@ -1,3 +1,3 @@
 version https://git-lfs.github.com/spec/v1
-oid sha256:c60bc8d172074586b6312d3bacea3e8285d9d20464136172936c4807e3333931
-size 8790
+oid sha256:1b5c3455d552c264a214626c46f8fde54be3663b07fd360040529a6e01fbcb9b
+size 9870