File size: 1,093 Bytes
a9f45b9 a28ddd5 a9f45b9 a28ddd5 a9f45b9 a28ddd5 a9f45b9 |
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 |
---
license: apache-2.0
tags:
- distilbart
- summarization
base_model: sshleifer/distilbart-cnn-12-6
model-index:
- name: MLQ-distilbart-bbc
results:
- task:
type: summarization
name: Summarization
dataset:
name: bbc
type: bbc
config: default
split: test
metrics:
- type: rouge
value: 61.43
name: ROUGE-2
verified: false
---
<!-- 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. -->
# MLQ-distilbart-bbc
This model is a fine-tuned version of [sshleifer/distilbart-cnn-12-6](https://huggingface.co/sshleifer/distilbart-cnn-12-6) on the BBC News Summary dataset (https://www.kaggle.com/pariza/bbc-news-summary).
The model has been generated as part of the in-lab practice of **Deep NLP course** currently held at Politecnico di Torino.
Training parameters:
- `num_train_epochs=2`
- `fp16=True`
- `per_device_train_batch_size=1`
- `warmup_steps=10`
- `weight_decay=0.01`
- `max_seq_length=100` |