Downloads · 30 days
178
0% of all-time downloads
ml6team/mt5-small-german-finetune-mlsum
mt5-small-german-finetune-mlsum is a summarization model from ml6team. Use it when you need a shorter version of a longer text. It is set up for transformers.
This model was finetuned for 3 epochs with a maxlen (input) of 768 tokens and targetmaxlen of 192 tokens. It was fine-tuned on all German articles present in the train split of the MLSUM dataset having less than 384 "…
Downloads · 30 days
178
0% of all-time downloads
All-time downloads
36.9K
Public
Repo size
3.6 GB
Likes
11
Public
Click a slice to open those files.
.h51.2 GB · 50%
From the Hugging Face model README
This model was finetuned for 3 epochs with a max_len (input) of 768 tokens and target_max_len of 192 tokens.
It was fine-tuned on all German articles present in the train split of the MLSUM dataset having less than 384 "words" after splitting on whitespace, which resulted in 80249 articles.
The exact expression to filter the dataset was the following:
dataset = dataset.filter(lambda e: len(e['text'].split()) < 384)
The fine-tuned model was evaluated on 2000 random articles from the validation set. Mean f1 ROUGE scores were calculated for both the fine-tuned model and the lead-3 baseline (which simply produces the leading three sentences of the document) and are presented in the following table.
| Model | Rouge-1 | Rouge-2 | Rouge-L |
|---|---|---|---|
| mt5-small | 0.399 | 0.318 | 0.392 |
| lead-3 | 0.343 | 0.263 | 0.341 |