File size: 1,120 Bytes
5c5ae5b
 
 
 
 
 
 
 
 
 
160a06d
 
2f070e7
 
 
 
 
 
 
 
 
75f6e82
2f070e7
 
 
 
 
 
 
 
75f6e82
2f070e7
 
 
 
 
 
 
 
75f6e82
 
2f070e7
 
5c5ae5b
 
 
 
 
 
 
 
 
 
 
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
---
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-Precision
            type: Precision
            value: 0.728
      - task:
          type: text-summarization
        dataset:
          name: liputan6
          type: canonical
        metrics:
          - name: BERTScore-Recall
            type: Recall
            value: 0.739
      - task:
          type: text-summarization
        dataset:
          name: liputan6
          type: canonical
        metrics:
          - name: BERTScore-F1
            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)