|
--- |
|
license: unknown |
|
datasets: |
|
- fajrikoto/id_liputan6 |
|
language: |
|
- id |
|
metrics: |
|
- rouge |
|
base_model: |
|
- cahya/t5-base-indonesian-summarization-cased |
|
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.525 |
|
- task: |
|
type: text-summarization |
|
dataset: |
|
name: liputan6 |
|
type: canonical |
|
metrics: |
|
- name: ROUGE-2 |
|
type: ROUGE-2 |
|
value: 0.448 |
|
- task: |
|
type: text-summarization |
|
dataset: |
|
name: liputan6 |
|
type: canonical |
|
metrics: |
|
- name: ROUGE-L |
|
type: ROUGE-L |
|
value: 0.491 |
|
--- |
|
|
|
model_path = "kairaamilanii/T5-ExtractiveSummarizer" |
|
|
|
summarizer = T5ForConditionalGeneration.from_pretrained(model_path) |
|
|
|
tokenizer = T5Tokenizer.from_pretrained(model_path) |