Update README.md

README.md

@@ -7,4 +7,28 @@ base_model:
 metrics:
 - accuracy
 pipeline_tag: text-classification
----
+
+# Sentiment Classification with Fine-Tuned BERT on Amazon Reviews
+
+This repository contains a fine-tuned BERT model for sentiment classification of Amazon product reviews The model classifies a given review into two classes: Positive and Negative
+
+---
+
+## **Model Overview**
+- **Base Model**: [google-bert/bert-base-uncased](https://huggingface.co/google-bert/bert-base-uncased)
+- **Dataset**: [SetFit/amazon_reviews_multi_en](https://huggingface.co/datasets/SetFit/amazon_reviews_multi_en),
+- **Classes**: Binary classification (`Positive`, `Negative`)
+- **Performance**:
+  - **Test Accuracy**: 89%
+  - **Validation Accuracy**: 89%
+
+*Figure 1: Confusion matrix for test data*
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6585ab80ef59559493941225/qBheOoyz8f_XD7BVFaRT9.png)
+
+*Figure 2: Confusion matrix for validation data*
+![image/png](https://cdn-uploads.huggingface.co/production/uploads/6585ab80ef59559493941225/Js_Bkuf9JDHuvZvgZJ61c.png)
+
+---
+
+
+