moazx commited on
Commit
2dea48e
·
verified ·
1 Parent(s): e51dfae

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +0 -79
README.md CHANGED
@@ -1,79 +0,0 @@
1
- ---
2
- license: mit
3
- pipeline_tag: text-classification
4
- ---
5
-
6
- # AraBERT-Restaurant-Sentiment
7
-
8
- This repository contains the fine-tuned model `moazx/AraBERT-Restaurant-Sentiment` for classifying Arabic restaurant reviews into positive and negative sentiments. The model is based on the AraBERT architecture and trained on a dataset of 800 Arabic restaurant reviews, collected and labeled using ChatGPT.
9
-
10
- ## Model Description
11
-
12
- The `AraBERT-Restaurant-Sentiment` model is fine-tuned to classify Arabic restaurant reviews into two categories:
13
- - Positive
14
- - Negative
15
-
16
- The dataset used for training consists of 400 positive and 400 negative reviews, covering multiple Arabic dialects.
17
-
18
- ## Sample Output
19
-
20
- Below are some examples of the model's output:
21
-
22
- ## Sample Output
23
-
24
- Below are some examples of the model's output:
25
-
26
- ### Example 1
27
-
28
- Input: المطعم ما عجبني، الطعم مو حلو والخدمة كانت سيئة جداً، والموظفين ما كانوا محترمين. الأسعار غالية مقارنة بالجودة. ما بنصح فيه.
29
-
30
- Expected Classification: سلبي
31
-
32
- Predicted Classification: سلبي
33
-
34
- Probability (Negative): 0.98
35
-
36
- Probability (Positive): 0.02
37
-
38
- ### Example 2
39
-
40
- Input: المطعم يجنن والاكل تحفة
41
-
42
- Expected Classification: إيجابي
43
-
44
- Predicted Classification: إيجابي
45
-
46
- Probability (Negative): 0.01
47
-
48
- Probability (Positive): 0.99
49
-
50
-
51
-
52
- ## Training and Dataset
53
-
54
- The model was trained using the notebook available at [Kaggle](https://www.kaggle.com/code/moazeldsokyx/arabert-arabic-sentiment-analysis). The training process involves the following steps:
55
-
56
- 1. Data Collection: 800 reviews (400 positive, 400 negative) were collected and labeled using ChatGPT.
57
- 2. Preprocessing: Text normalization, tokenization, and other preprocessing steps were applied.
58
- 3. Fine-Tuning: The AraBERT model was fine-tuned on the prepared dataset.
59
- 4. Evaluation: The model was evaluated using accuracy and other relevant metrics to ensure its performance.
60
-
61
- ## Usage
62
-
63
- To use the `moazx/AraBERT-Restaurant-Sentiment` model, you can load it using the Hugging Face `transformers` library. Below is an example of how to use the model to classify a restaurant review:
64
-
65
-
66
- ```python
67
- from transformers import AutoTokenizer, AutoModelForSequenceClassification
68
- import torch
69
-
70
- # Load the tokenizer and model
71
- tokenizer = AutoTokenizer.from_pretrained("moazx/AraBERT-Restaurant-Sentiment")
72
- model = AutoModelForSequenceClassification.from_pretrained("moazx/AraBERT-Restaurant-Sentiment")
73
- ```
74
-
75
- ## Acknowledgements
76
-
77
- The dataset used for training the model was collected and labeled using ChatGPT. Special thanks to the creators of AraBERT and the Hugging Face team for their continuous support and development of open-source NLP tools.
78
-
79
- For more details on the training process and dataset, please refer to the [Kaggle Notebook](https://www.kaggle.com/code/moazeldsokyx/arabert-arabic-sentiment-analysis).