Downloads · 30 days
15
1% of all-time downloads
ndtran/t5-small_cnn-daily-mail
t5-small_cnn-daily-mail is a machine learning model from ndtran. Use it for the machine learning task on the model card, and read the license before you ship it in a product. It is set up for transformers.
We are focusing on Abstractive Text Summarization. Briefly, the input of the task is a text paragraph and output is a summarization of the input which is similar to the input from its meaning. Compared to another appr…
Downloads · 30 days
15
1% of all-time downloads
All-time downloads
2.1K
Public
Parameters
60.5M
484 MB on disk
Likes
0
Public
Click a slice to open those files.
.bin242 MB · 50%
From the Hugging Face model README
We are focusing on Abstractive Text Summarization. Briefly, the input of the task is a text paragraph and output is a summarization of the input which is similar to the input from its meaning. Compared to another approach (Extractive Text Summarization), Abstractive Text Summarization is outstanding in the output quality.
The model was finetuned on CNN/DailyMail. That is an English-language dataset containing just over 300k unique news articles as written by journalists at CNN and the Daily Mail. The current version supports both extractive and abstractive summarization, though the original version was created for machine reading and comprehension and abstractive question answering.
| Parameter | Value |
|---|---|
| No. Epoch | 3 |
| Learning rate | 1e-5 (First two epochs), 5e-6 (Last epoch) |
| Optimizer | AdamW |
| Layers | Full |
All of those metrics are from the evaluation of the finetuned model on the test-set of CNN/DailyMail.
| Metrics | Recall | Precision | F1-Score |
|---|---|---|---|
| Rouge 1 | 0.38 | 0.42 | 0.39 |
| Rouge 2 | 0.16 | 0.18 | 0.17 |
| Rouge L | 0.27 | 0.3 | 0.27 |
| Rouge L-Sum | 0.27 | 0.3 | 0.27 |
license: apache-2.0 datasets: