Downloads · 30 days
3
6% of all-time downloads
safiati/arabertv2-multiclass-model
arabertv2-multiclass-model is a text classification model from safiati. Use it when you need a label for a piece of text. The card lists the license as apache-2.0.
تم تدريب هذا النموذج اعتمادًا على AraBERTv2 وإعادة تدريبه على بيانات متعددة الفئات لتصنيف النصوص العربية (وبالأخص الدارجة الجزائرية) ضمن عدّة تصنيفات:
Downloads · 30 days
3
6% of all-time downloads
All-time downloads
53
Public
Parameters
135M
541 MB on disk
Likes
0
Public
Click a slice to open those files.
.safetensors541 MB · 100%
From the Hugging Face model README
تم تدريب هذا النموذج اعتمادًا على AraBERTv2 وإعادة تدريبه على بيانات متعددة الفئات لتصنيف النصوص العربية (وبالأخص الدارجة الجزائرية) ضمن عدّة تصنيفات:
يستخدم النموذج للكشف عن الخطاب المسيء وتصفية المحتوى على منصات التواصل.
| الخاصية | القيمة |
|---|---|
| النموذج الأساسي | aubmindlab/bert-base-arabertv02 |
| عدد الحقبات (epochs) | 5 |
| حجم الدفعة (batch size) | 8 |
| طول النص الأقصى | 128 |
| معدل التعلم | 2e-5 |
| optimizer | AdamW |
| خسارة التدريب | CrossEntropyLoss |
| طريقة التقسيم | train/test split بنسبة 20% |
| عدد الفئات | متعدد الفئات (multiclass) |
| المقياس | القيمة |
|---|---|
| Accuracy | 0.6163 |
| Precision | 0.6221 |
| Recall | 0.6163 |
| F1-score | 0.6175 |
يمكنكِ إضافة نتائجك من Colab مباشرة تحت هذا الجدول.
from transformers import AutoTokenizer, AutoModelForSequenceClassification
import torch
model_name = "safiati/arabertv2-multiclass-model"
tokenizer = AutoTokenizer.from_pretrained(model_name)
model = AutoModelForSequenceClassification.from_pretrained(model_name)
text = "جايح ومش مربي"
inputs = tokenizer(text, return_tensors="pt")
with torch.no_grad():
logits = model(**inputs).logits
probs = torch.softmax(logits, dim=1)
pred_id = torch.argmax(probs).item()
print("Prediction:", model.config.id2label[pred_id])