Views
No views yet
1---
2language:
3- my
4license: mit
5library_name: transformers
6pipeline_tag: text2text-generation
7tags:
8- Burmese
9- Myanmar
10- sports
11- question-answer-generation
12- mT5
13- text2text-generation
14---
15
16# mT5-base Burmese Sports QA Generation
17
18A fine-tuned `google/mt5-base` model for generating question-answer pairs from Burmese sports news articles.
19
20## Model Description
21
22This model was fine-tuned for a university NLP project using a dataset of Burmese sports news articles and generated question-answer pairs.
23
24The task is **text-to-text question-answer generation**, not conversational question answering or chatbot generation.
25
26### Input
27
28A Burmese sports news article.
29
30### Output
31
32A generated question-answer pair in Burmese.
33
34Example format:
35
36```text
37မေးခွန်း: ...
38အဖြေ: ...| Split | Articles | QA Pairs |
|---|---|---|
| Training | 4,515 | 21,049 |
| Validation | 564 | 2,614 |
| Test | 565 | 2,616 |
| Total | 5,644 | 26,279 |
| Parameter | Value |
|---|---|
| Base model | google/mt5-base |
| Epochs | 10 |
| Learning rate | 3e-5 |
| Training batch size | 2 |
| Evaluation batch size | 2 |
| Gradient accumulation | 8 |
| Effective batch size | 16 |
| Weight decay | 0.01 |
| Warmup ratio | 0.1 |
| Label smoothing | 0.1 |
| Optimizer | Adafactor |
| FP16 | Disabled |
| Gradient checkpointing | Enabled |
pip install transformers sentencepiece torch1from transformers import AutoTokenizer, AutoModelForSeq2SeqLM
2
3model_name = "Kaun9MyatHt3t/mt5-base-burmese-sports-qa"
4
5tokenizer = AutoTokenizer.from_pretrained(model_name)
6model = AutoModelForSeq2SeqLM.from_pretrained(model_name)
7
8article = "မြန်မာ့လက်ရွေးစင်အသင်းသည် ယနေ့ ဘောလုံးပွဲတစ်ပွဲကို ယှဉ်ပြိုင်ကစားခဲ့သည်။"
9
10inputs = tokenizer(
11 article,
12 return_tensors="pt",
13 truncation=True,
14 max_length=256
15)
16
17outputs = model.generate(
18 **inputs,
19 max_new_tokens=128,
20 num_beams=4
21)
22
23result = tokenizer.decode(
24 outputs[0],
25 skip_special_tokens=True
26)
27
28print(result)google/mt5-base