File size: 2,468 Bytes
c9be42a fa3e6d9 c9be42a 8b878fb c9be42a c6163a6 fa3e6d9 c6163a6 fa3e6d9 c9be42a c6163a6 fa3e6d9 c9be42a 8b878fb c6163a6 c9be42a 8b878fb |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 48 49 50 51 52 53 54 55 56 57 58 59 60 61 62 63 64 65 66 67 68 |
---
base_model: google/pegasus-cnn_dailymail
tags:
- generated_from_trainer
datasets:
- xlsum
model-index:
- name: pegsasus_xlsum
results: []
language:
- en
metrics:
- rouge
---
<!-- 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. -->
# my_awesome_pegsasus_model
This model is a fine-tuned version of [google/pegasus-cnn_dailymail](https://huggingface.co/google/pegasus-cnn_dailymail) on the xlsum dataset.
## Model description
**pegasus_xlsum** is a state-of-the-art model fine-tuned on the **English** subset of the **csebuetnlp/xlsum** dataset. This data source is one of the most comprehensive and diverse sets available, originally composed of **1.35 million** professional article-summary pairs sourced from BBC across 45 languages. Despite its multilingual nature, we intentionally selected the **English** language subset, consisting of approximately **330,000** records, as the focus for our fine-tuning process.
The goal was to adapt the model for the text summarization task, and we're thrilled to report that the fine-tuned **pegasus_xlsum** model exceeded our expectations. It outperformed the established **csebuetnlp/mT5_multilingual_XLSum** model in terms of **ROUGE** scores, demonstrating superior summary generation capabilities. The **pegasus_xlsum** model leverages the powerful PEGASUS architecture, proving its efficiency and effectiveness in handling **English** text summarization tasks.
## Intended uses & limitations
**pegasus_xlsum** is to provide a reliable, high-performance solution for **English** text summarization, making the most of the rich, professional, and diverse source dataset it was trained on. We hope you find this model as useful in your applications as we did in our experiments.
## Training and evaluation data
| Metric | Score |
|--------|-------|
| **ROUGE-1 F1** | 0.391215 |
| **ROUGE-2 F1** | 0.174674 |
| **ROUGE-L F1** | 0.308941 |
| **ROUGE-LSUM F1** | 0.308923 |
## Training procedure
### Training hyperparameters
The following hyperparameters were used during training:
- learning_rate: 2e-05
- train_batch_size: 8
- eval_batch_size: 8
- seed: 42
- optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
- lr_scheduler_type: linear
- num_epochs: 4
### Framework versions
- Transformers 4.31.0
- Pytorch 2.0.1+cu118
- Datasets 2.14.0
- Tokenizers 0.13.3 |