Update README.md
Browse files
README.md
CHANGED
|
@@ -13,7 +13,7 @@ Unlike standard pre-trained models (like BERT), this architecture was built from
|
|
| 13 |
|
| 14 |
* **Architecture**: 4-Layer Transformer Encoder
|
| 15 |
* **Task**: Binary Sentiment Analysis (Positive/Negative)
|
| 16 |
-
* **Accuracy**: 89.
|
| 17 |
* **Parameters**: Optimized for efficient inference on edge devices
|
| 18 |
|
| 19 |
## 🛠️ Technical Specifications
|
|
@@ -26,6 +26,7 @@ Unlike standard pre-trained models (like BERT), this architecture was built from
|
|
| 26 |
## 💻 Training Environment
|
| 27 |
This model was trained locally on an **Apple Mac mini M4** with **24GB of Unified Memory**.
|
| 28 |
* **Accelerator**: Metal Performance Shaders (MPS)
|
|
|
|
| 29 |
* **Dataset**: Subset of 500,000 samples from Amazon Polarity
|
| 30 |
|
| 31 |
## 📈 Performance & Insights
|
|
|
|
| 13 |
|
| 14 |
* **Architecture**: 4-Layer Transformer Encoder
|
| 15 |
* **Task**: Binary Sentiment Analysis (Positive/Negative)
|
| 16 |
+
* **Accuracy**: 89.73% on Test Set
|
| 17 |
* **Parameters**: Optimized for efficient inference on edge devices
|
| 18 |
|
| 19 |
## 🛠️ Technical Specifications
|
|
|
|
| 26 |
## 💻 Training Environment
|
| 27 |
This model was trained locally on an **Apple Mac mini M4** with **24GB of Unified Memory**.
|
| 28 |
* **Accelerator**: Metal Performance Shaders (MPS)
|
| 29 |
+
* **Training Time**: ~1.5 hours
|
| 30 |
* **Dataset**: Subset of 500,000 samples from Amazon Polarity
|
| 31 |
|
| 32 |
## 📈 Performance & Insights
|