fitsblb commited on
Commit
471d0d2
·
verified ·
1 Parent(s): 84f1605

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +49 -0
README.md CHANGED
@@ -0,0 +1,49 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ language: en
3
+ license: mit
4
+ tags:
5
+ - sentiment-analysis
6
+ - text-classification
7
+ - yelp
8
+ - transformers
9
+ - distilbert
10
+ ---
11
+
12
+ # Yelp Reviews Sentiment Analyzer
13
+
14
+ ## Model Overview
15
+ This is a DistilBERT-based sentiment analysis model fine-tuned on a subset of the Yelp Open Dataset. It classifies restaurant reviews into three categories: Negative, Neutral, and Positive.
16
+
17
+ ## Intended Use
18
+ - Sentiment classification of restaurant reviews for business insights, customer feedback analysis, or academic research.
19
+ - Can be integrated into applications to provide real-time sentiment detection.
20
+
21
+ ## Training Data
22
+ - Yelp Open Dataset (restaurant reviews subset).
23
+ - Labels derived from star ratings converted into sentiment classes.
24
+
25
+ ## Model Architecture
26
+ - Based on `distilbert-base-uncased`.
27
+ - Fine-tuned using Hugging Face's `AutoModelForSequenceClassification`.
28
+
29
+ ## Performance
30
+ - Accuracy: ~78.5%
31
+ - F1 Score: ~78.4%
32
+ - Precision: ~78.3%
33
+ - Recall: ~78.5%
34
+
35
+ ## Limitations
36
+ - Performance may vary on reviews from domains outside Yelp restaurants.
37
+ - Model is trained only on English-language reviews.
38
+ - Neutral class can be subjective, and borderline cases may be misclassified.
39
+
40
+ ## How to Use
41
+ Use Hugging Face Transformers pipeline:
42
+
43
+ ```python
44
+ from transformers import pipeline
45
+
46
+ sentiment_analyzer = pipeline("sentiment-analysis", model="fitsblb/YelpReviewsAnalyzer")
47
+
48
+ result = sentiment_analyzer("The food was amazing but the service was slow.")
49
+ print(result)