add README

This commit is contained in:
Rémi Louf
2019-12-06 11:49:27 +01:00
committed by Julien Chaumond
parent ade3cdf5ad
commit c0707a85d2
2 changed files with 61 additions and 17 deletions

View File

@@ -713,20 +713,3 @@ Training with the previously defined hyper-parameters yields the following resul
```bash
acc = 0.7093812375249501
```
### Abstractive Summarization
This example provides a simple API for the [BertAbs](https://github.com/nlpyang/PreSumm) model finetuned on the CNN/DailyMail dataset. The script can be used to generate summaries from any text.
```bash
python run_summarization.py \
--documents_dir 'path/to/documents' \
--summaries_output_dir 'path/to/summaries' \
--visible_gpus 0,1,2 \
--batch_size 4 \
--min_length 50 \
--max_length 200 \
--beam_size 5 \
--alpha 0.95 \
--block_trigram true
```