|
--- |
|
license: unknown |
|
datasets: |
|
- fajrikoto/id_liputan6 |
|
language: |
|
- id |
|
metrics: |
|
- rouge |
|
base_model: |
|
- cahya/bert2bert-indonesian-summarization |
|
pipeline_tag: summarization |
|
|
|
model-index: |
|
- name: BERT2BERT-Summarizer |
|
results: |
|
- task: |
|
type: text-summarization |
|
dataset: |
|
name: liputan6 |
|
type: canonical |
|
metrics: |
|
- name: ROUGE-1 |
|
type: ROUGE-1 |
|
value: 0.484 |
|
- task: |
|
type: text-summarization |
|
dataset: |
|
name: liputan6 |
|
type: canonical |
|
metrics: |
|
- name: ROUGE-2 |
|
type: ROUGE-2 |
|
value: 0.399 |
|
- task: |
|
type: text-summarization |
|
dataset: |
|
name: liputan6 |
|
type: canonical |
|
metrics: |
|
- name: ROUGE-L |
|
type: ROUGE-L |
|
value: 0.447 |
|
|
|
--- |
|
|
|
|
|
model_name = "kairaamilanii/BERT2BERT-ExtractiveSummarizer" |
|
|
|
tokenizer = BertTokenizer.from_pretrained(model_name) |
|
|
|
tokenizer.bos_token = tokenizer.cls_token |
|
|
|
tokenizer.eos_token = tokenizer.sep_token |
|
|
|
model = EncoderDecoderModel.from_pretrained(model_name) |
