Downloads · 30 days
9
1% of all-time downloads
Bert-hard-balanced
mofawzy/Bert-hard-balanced
Bert-hard-balanced 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 Hotel Arabic Reviews dataset from booking.com (HARD) dataset balanced version to identify sentiments opinion in Arabic language.
All-time downloads
837
Public
Repo size
9.4 GB
Likes
1
Public
Repo makeup
Click a slice to open those files.
.pt2.7 GB · 67%
At a glance
- Task
- Text Classification
- Library
- transformers
- Model type
- bert
- Access
- Public
- Created
- Mar 2, 2022
- Updated
- Feb 18, 2022
- SHA
- 1cfc9dc5
Try a prompt
Datasets
Model card
From the Hugging Face model README
BERT-ASTD Balanced
Arabic version bert model fine tuned on Hotel Arabic Reviews dataset from booking.com (HARD) dataset balanced version to identify sentiments opinion in Arabic language.
Data
The model were fine-tuned on ~93000 book reviews in arabic using bert large arabic
Dataset:
- Train 70%
- Validation: 10%
- Test: 20%
Results
| class | precision | recall | f1-score | Support |
|---|---|---|---|---|
| 0 | 0.9733 | 0.9547 | 0.9639 | 10570 |
| 1 | 0.9555 | 0.9738 | 0.9646 | 10570 |
| Accuracy | 0.9642 | 21140 |
How to use
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-hard-balanced"
model = AutoModelForSequenceClassification.from_pretrained(model_name,num_labels=2)
tokenizer = AutoTokenizer.from_pretrained(model_name)
- Task
- Text Classification
- Library
- transformers
- Type
- bert
- Languages
- ar
- Created
- Mar 2, 2022
- Updated
- Feb 18, 2022