mofawzy commited on
Commit
bdcf557
·
1 Parent(s): c67b820

update model card

Browse files
Files changed (1) hide show
  1. README.md +37 -1
README.md CHANGED
@@ -1 +1,37 @@
1
- Fine-tune BERT model on Large Book Reviews Arabic Dataset
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ----
2
+ language: ar
3
+ datasets:
4
+ - labr
5
+ tags:
6
+ - labr
7
+ ---
8
+
9
+
10
+ # BERT-LABR unbalanced
11
+ Arabic version bert model fine tuned on LABR dataset
12
+
13
+ ## Data
14
+ The model were fine-tuned on ~63000 book reviews in arabic using bert large arabic
15
+
16
+
17
+ ## Results
18
+ | class | precision | recall | f1-score | Support |
19
+ |----------|-----------|--------|----------|---------|
20
+ | 0 | 0.8109 | 0.6832 | 0.7416 | 1670 |
21
+ | 1 | 0.9399 | 0.9689 | 0.9542 | 8541 |
22
+ | Accuracy | | | 0.9221 | 10211 |
23
+
24
+
25
+
26
+ ## How to use
27
+
28
+ 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:
29
+
30
+ ```python
31
+ from transformers import AutoModelForSequenceClassification, AutoTokenizer
32
+
33
+ model_name="mofawzy/bert-labr-unbalanced"
34
+ model = AutoModelForSequenceClassification.from_pretrained(model_name,num_labels=2)
35
+ tokenizer = AutoTokenizer.from_pretrained(model_name)
36
+
37
+ ```