Downloads · 30 days
14
2% of all-time downloads
mofawzy/bert-labr-unbalanced
bert-labr-unbalanced is a text classification model from mofawzy. Use it when you need a label for a piece of text. It is set up for transformers.
Arabic version bert model fine tuned on LABR dataset
Downloads · 30 days
14
2% of all-time downloads
All-time downloads
932
Public
Parameters
337M
6.7 GB on disk
Likes
0
Public
Click a slice to open those files.
.pt2.7 GB · 40%
How the weights are stored.
F32337M · 100%
From the Hugging Face model README
Arabic version bert model fine tuned on LABR dataset
The model were fine-tuned on ~63000 book reviews in arabic using bert large arabic
| class | precision | recall | f1-score | Support |
|---|---|---|---|---|
| 0 | 0.8109 | 0.6832 | 0.7416 | 1670 |
| 1 | 0.9399 | 0.9689 | 0.9542 | 8541 |
| Accuracy | 0.9221 | 10211 |
You can use these models by installing torch or tensorflow and Huggingface library transformers. And you can use it directly by initializing it like this:
from transformers import AutoModelForSequenceClassification, AutoTokenizer
model_name="mofawzy/bert-labr-unbalanced"
model = AutoModelForSequenceClassification.from_pretrained(model_name,num_labels=2)
tokenizer = AutoTokenizer.from_pretrained(model_name)