Update README.md
Browse files
README.md
CHANGED
@@ -13,6 +13,40 @@ tags:
|
|
13 |
|
14 |
# Model Card for Model ID
|
15 |
|
16 |
-
|
17 |
-
The mT5-multilingual-XLSum model was fine-tuned on the UA-News dataset to generate news headlines in Ukrainian language.
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
18 |
|
|
|
13 |
|
14 |
# Model Card for Model ID
|
15 |
|
16 |
+
## Model Summary
|
17 |
+
The mT5-multilingual-XLSum model was fine-tuned on the UA-News dataset to generate news headlines in Ukrainian language. This model is designed to produce concise and accurate summaries for Ukrainian news articles.
|
18 |
+
|
19 |
+
## Model Details
|
20 |
+
- **Model Type**: Summarization
|
21 |
+
- **Language**: Multilingual
|
22 |
+
- **Library**: Transformers
|
23 |
+
|
24 |
+
## Dataset
|
25 |
+
- **Name**: UA-News
|
26 |
+
- **Source**: FIdo-AI
|
27 |
+
- **Description**: The UA-News dataset contains a diverse collection of Ukrainian news articles, covering various topics including politics, economics, culture, and sports. The dataset is curated to provide high-quality training data for summarization tasks.
|
28 |
+
|
29 |
+
## Training
|
30 |
+
- **Fine-Tuning**: The model was fine-tuned on the UA-News dataset using the Hugging Face Transformers library.
|
31 |
+
- **Epochs**: 4
|
32 |
+
- **Batch Size**: 4
|
33 |
+
- **Learning Rate**: 4e-5
|
34 |
+
|
35 |
+
## Evaluation
|
36 |
+
- **Metrics**: The model's performance was evaluated using the ROUGE metric.
|
37 |
+
- **ROUGE-1**: 0.2452
|
38 |
+
- **ROUGE-2**: 0.1075
|
39 |
+
- **ROUGE-L**: 0.2348
|
40 |
+
- **BERTScore**: 0.7573
|
41 |
+
|
42 |
+
## Usage
|
43 |
+
- **Pipeline Tag**: Summarization
|
44 |
+
- **How to Use**: The model can be used with the Hugging Face `pipeline` for summarization. Here's an example:
|
45 |
+
```python
|
46 |
+
from transformers import pipeline
|
47 |
+
|
48 |
+
summarizer = pipeline("summarization", model="yelyah/mT5-XLSUM-ua-news ")
|
49 |
+
article = "Your news article text here."
|
50 |
+
summary = summarizer(article)
|
51 |
+
print(summary)
|
52 |
|