Views
No views yet
bert-base-multilingual-cased| Task | Metric | Score |
|---|---|---|
| Crash Detection (Binary) | Accuracy | 91.1% |
1from transformers import AutoModelForSequenceClassification, AutoTokenizer
2
3model = AutoModelForSequenceClassification.from_pretrained("crash-information-extraction/BERT-Crash-nonCrash-Classification")
4tokenizer = AutoTokenizer.from_pretrained("crash-information-extraction/BERT-Crash-nonCrash-Classification")
5
6text = "تصادف در خیابان آزادی ۲ کشته داشت"
7inputs = tokenizer(text, return_tensors="pt")
8outputs = model(**inputs)
9predicted_class = outputs.logits.argmax().item()