mohamedadaly/labr
Viewer • Updated • 14.7k • 592 • 3
How to use mofawzy/bert-labr-unbalanced with Transformers:
# Use a pipeline as a high-level helper
from transformers import pipeline
pipe = pipeline("text-classification", model="mofawzy/bert-labr-unbalanced") # Load model directly
from transformers import AutoTokenizer, AutoModelForSequenceClassification
tokenizer = AutoTokenizer.from_pretrained("mofawzy/bert-labr-unbalanced")
model = AutoModelForSequenceClassification.from_pretrained("mofawzy/bert-labr-unbalanced")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)