metadata
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)