transformers pytorch tensorboard t5 text2text-generation generated_from_trainer license:apache-2.0 autotrain_compatible endpoints_compatible text-generation-inference region:us
---
license: apache-2.0
tags:
- generated_from_trainer
metrics:
- rouge
model-index:
- name: t5_recommendation_sports_equipment_english
  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_recommendation_sports_equipment_english

This model is a fine-tuned version of [t5-large](https://huggingface.co/t5-large) on the None dataset.
It achieves the following results on the evaluation set:
- Loss: 0.3633
- Rouge1: 59.2063
- Rouge2: 52.3810
- Rougel: 59.0476
- Rougelsum: 59.3651
- Gen Len: 3.8571

## 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: 0.0001
- train_batch_size: 4
- eval_batch_size: 4
- seed: 42
- gradient_accumulation_steps: 4
- total_train_batch_size: 16
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 10

### Training results

| Training Loss | Epoch | Step | Validation Loss | Rouge1  | Rouge2  | Rougel  | Rougelsum | Gen Len |
|:-------------:|:-----:|:----:|:---------------:|:-------:|:-------:|:-------:|:---------:|:-------:|
| No log        | 0.96  | 6    | 7.0808          | 15.2846 | 5.0505  | 15.1292 | 15.1867   | 18.6190 |
| No log        | 1.96  | 12   | 3.2252          | 14.6825 | 0.0     | 14.4444 | 14.9206   | 4.7619  |
| No log        | 2.96  | 18   | 0.8665          | 7.1429  | 0.0     | 7.1429  | 7.1429    | 3.8095  |
| No log        | 3.96  | 24   | 0.6607          | 42.8571 | 33.3333 | 42.7778 | 43.1746   | 3.3333  |
| No log        | 4.96  | 30   | 0.5453          | 43.6508 | 28.5714 | 42.8571 | 43.6508   | 3.7143  |
| No log        | 5.96  | 36   | 0.5628          | 52.6984 | 42.8571 | 51.9048 | 52.5397   | 3.8571  |
| No log        | 6.96  | 42   | 0.4353          | 46.0317 | 38.0952 | 45.2381 | 46.0317   | 3.8571  |
| No log        | 7.96  | 48   | 0.3928          | 54.4444 | 47.6190 | 54.2857 | 54.7619   | 3.8095  |
| No log        | 8.96  | 54   | 0.3641          | 61.9048 | 57.1429 | 62.6984 | 62.6984   | 3.9048  |
| No log        | 9.96  | 60   | 0.3633          | 59.2063 | 52.3810 | 59.0476 | 59.3651   | 3.8571  |


### Framework versions

- Transformers 4.26.0
- Pytorch 2.1.0+cu121
- Datasets 2.8.0
- Tokenizers 0.13.3
