Update README.md

README.md

@@ -16,18 +16,18 @@ tags:
 ---
 # Yelp Review Classifier
 
-This model is a sentiment classification model for Yelp reviews, trained to predict whether a review is **positive** or **negative**. The model was fine-tuned using the `distilbert-base-uncased` model architecture, based on the [DistilBERT model](https://huggingface.co/distilbert/distilbert-base-uncased) from Hugging Face, and trained on a Yelp reviews dataset.
+This model is a sentiment classification model for Yelp reviews, trained to predict whether a review is **star ratings (1 to 5 stars)**. The model was fine-tuned using the `distilbert-base-uncased` model architecture, based on the [DistilBERT model](https://huggingface.co/distilbert/distilbert-base-uncased) from Hugging Face, and trained on a Yelp reviews dataset.
 
 ## Model Details
 - **Model Type**: DistilBERT-based model for sequence classification
 - **Model Architecture**: `distilbert-base-uncased`
 - **Number of Parameters**: Approximately 66M parameters
-- **Training Dataset**: The model was trained on a curated Yelp reviews dataset, labeled for sentiment (positive/negative).
-- **Fine-Tuning Task**: Sentiment analysis for Yelp reviews (positive or negative sentiment)
+- **Training Dataset**: The model was trained on a curated Yelp reviews dataset, labeled for star ratings (1 to 5 stars).
+- **Fine-Tuning Task**: Multi-class classification for Yelp reviews, predicting the star rating (from 1 to 5 stars) based on the content of the review.
 
 ## Training Data
 - **Dataset**: Custom Yelp reviews dataset
-- **Data Description**: The dataset consists of Yelp reviews, each labeled with a sentiment (positive/negative).
+- **Data Description**: The dataset consists of Yelp reviews, labeled for star ratings (1 to 5 stars).
 - **Preprocessing**: The dataset was preprocessed by cleaning the reviews to remove unwanted characters and URLs.
 
 ## Training Details