Views
No views yet
| Metric | Score |
|---|---|
| Accuracy | 90.00% |
| Precision | 91.33% |
| Recall | 100.00% |
| F1-Score | 90.91% |
1from transformers import AutoTokenizer, AutoModelForSequenceClassification
2import torch
3
4tokenizer = AutoTokenizer.from_pretrained("Silxxor/russian-turn-detector")
5model = AutoModelForSequenceClassification.from_pretrained("Silxxor/russian-turn-detector")
6
7text = "да я понимаю что"
8inputs = tokenizer(text, return_tensors="pt", max_length=64, truncation=True, padding=True)
9
10with torch.no_grad():
11 outputs = model(**inputs)
12 prediction = torch.argmax(outputs.logits, dim=-1).item()
13
14# 0 = incomplete turn, 1 = complete turn
15print("Complete" if prediction == 1 else "Incomplete")