pytorch bart seq2seq en dataset:samsum transformers azureml summarization license:apache-2.0 text2text-generation
.gitattributes README.md config.json merges.txt pytorch_model.bin special_tokens_map.json tokenizer.json tokenizer_config.json vocab.json

---
language: en
tags:
- azureml
- bart
- summarization
license: apache-2.0
datasets:
- samsum
model-index:
- name: bart-large-samsum
  results:
  - task: 
      name: Abstractive Text Summarization
      type: abstractive-text-summarization
    dataset:
      name: "SAMSum Corpus: A Human-annotated Dialogue Dataset for Abstractive Summarization" 
      type: samsum
widget:
- text: | 
    Henry: Hey, is Nate coming over to watch the movie tonight?
    Kevin: Yea, he said he'll be arriving a bit later at around 7 since he gets off of work at 6. Have you taken out the garbage yet?
    Henry: Oh I forgot. I'll do that once I'm finished with my assignment for my math class.
    Kevin: Yea, you should take it out as soon as possible. And also, Nate is bringing his girlfriend.
    Henry: Nice, I'm really looking forward to seeing them again.
---

## `bart-large-samsum`

This model was trained using `AzureML`. It was fine-tuned on the `SAMSum` corpus from `facebook/bart-large` checkpoint.


## Usage
```python
from transformers import pipeline
summarizer = pipeline("summarization", model="henryu-lin/bart-large-samsum")

conversation = '''Henry: Hey, is Nate coming over to watch the movie tonight?
    Kevin: Yea, he said he'll be arriving a bit later at around 7 since he gets off of work at 6. Have you taken out the garbage yet?
    Henry: Oh I forgot. I'll do that once I'm finished with my assignment for my math class.
    Kevin: Yea, you should take it out as soon as possible. And also, Nate is bringing his girlfriend.
    Henry: Nice, I'm really looking forward to seeing them again.
'''
summarizer(conversation)
```

## Results
| key | value |
| --- | ----- |
| eval_rouge1 | 54.5013 |
| eval_rouge2 | 29.3672 |
| eval_rougeL | 44.784 |
| eval_rougeLsum | 49.906 |
| predict_rouge1 | 53.5417 |
| predict_rouge2 | 28.3906 |
| predict_rougeL | 44.2561 |
| predict_rougeLsum | 49.1162 |
