---
language: uk

datasets: Yehor/ual-topics

license: cc-by-nc-sa-4.0
---

## Metrics

- Epochs: 50
- Test accuracy: 0.62500
- F1 (MACRO): 0.46573

## How to use

```python
import torch
from transformers import AutoTokenizer, AutoModel

tokenizer = AutoTokenizer.from_pretrained("Yehor/ual-topics-classifier")
model = AutoModel.from_pretrained("Yehor/ual-topics-classifier")

device = torch.device('cuda' if torch.cuda.is_available() else 'cpu')

topic_classifier = pipeline(task='text-classification', model=model, tokenizer=tokenizer, device=device)

question = """
Що мені робити на ВЛК
"""

print(topic_classifier(question))

question = """
Які мої дії для отримання аліментів на дитину
"""

print(topic_classifier(question))
```