Update README.md
Browse files
README.md
CHANGED
@@ -95,6 +95,8 @@ for text, result in zip(texts, results):
|
|
95 |
|
96 |
## π Performance Overview
|
97 |
|
|
|
|
|
98 |
### Benchmark Results
|
99 |
|
100 |
| Dataset | Pre-trained RoBERTa | RoBERTa-Sentimentic | Improvement |
|
@@ -113,6 +115,8 @@ for text, result in zip(texts, results):
|
|
113 |
|
114 |
## π― Model Performance Analysis
|
115 |
|
|
|
|
|
116 |
### Confusion Matrices
|
117 |
|
118 |
#### IMDB Dataset Results
|
@@ -146,6 +150,8 @@ Precision: 91.5% | Recall: 91.4% | F1-Score: 91.5%
|
|
146 |
|
147 |
## π οΈ Technical Details
|
148 |
|
|
|
|
|
149 |
### Architecture
|
150 |
- **Base Model**: [roberta-base](https://huggingface.co/roberta-base) (125M parameters)
|
151 |
- **Task Head**: Linear classification layer with dropout (0.1)
|
@@ -183,6 +189,8 @@ Training Time: ~25 minutes total
|
|
183 |
|
184 |
## π Training Process
|
185 |
|
|
|
|
|
186 |
### Phase 1: IMDB Fine-tuning
|
187 |
- **Dataset**: 25,000 IMDB movie reviews
|
188 |
- **Strategy**: Same-domain fine-tuning
|
|
|
95 |
|
96 |
## π Performance Overview
|
97 |
|
98 |
+

|
99 |
+
|
100 |
### Benchmark Results
|
101 |
|
102 |
| Dataset | Pre-trained RoBERTa | RoBERTa-Sentimentic | Improvement |
|
|
|
115 |
|
116 |
## π― Model Performance Analysis
|
117 |
|
118 |
+

|
119 |
+
|
120 |
### Confusion Matrices
|
121 |
|
122 |
#### IMDB Dataset Results
|
|
|
150 |
|
151 |
## π οΈ Technical Details
|
152 |
|
153 |
+

|
154 |
+
|
155 |
### Architecture
|
156 |
- **Base Model**: [roberta-base](https://huggingface.co/roberta-base) (125M parameters)
|
157 |
- **Task Head**: Linear classification layer with dropout (0.1)
|
|
|
189 |
|
190 |
## π Training Process
|
191 |
|
192 |
+

|
193 |
+
|
194 |
### Phase 1: IMDB Fine-tuning
|
195 |
- **Dataset**: 25,000 IMDB movie reviews
|
196 |
- **Strategy**: Same-domain fine-tuning
|