|
--- |
|
license: unknown |
|
datasets: |
|
- fajrikoto/id_liputan6 |
|
language: |
|
- id |
|
metrics: |
|
- rouge |
|
base_model: |
|
- cahya/bert2bert-indonesian-summarization |
|
|
|
|
|
model-index: |
|
- name: BERT2BERT-Summarizer |
|
results: |
|
- task: |
|
type: text-summarization |
|
dataset: |
|
name: liputan6 |
|
type: canonical |
|
metrics: |
|
- name: BERTScore |
|
type: Precision |
|
value: 0.728 |
|
- task: |
|
type: text-summarization |
|
dataset: |
|
name: liputan6 |
|
type: canonical |
|
metrics: |
|
- name: BERTScore |
|
type: Recall |
|
value: 0.739 |
|
- task: |
|
type: text-summarization |
|
dataset: |
|
name: liputan6 |
|
type: canonical |
|
metrics: |
|
- name: BERTScore |
|
type: f1 |
|
value: 0.733 |
|
|
|
--- |
|
|
|
model_name = "kairaamilanii/BERT2BERT-AbstractiveSummarizer" |
|
|
|
tokenizer = BertTokenizer.from_pretrained(model_name) |
|
|
|
tokenizer.bos_token = tokenizer.cls_token |
|
|
|
tokenizer.eos_token = tokenizer.sep_token |
|
|
|
model = EncoderDecoderModel.from_pretrained(model_name) |
