Update README.md
Browse files
README.md
CHANGED
|
@@ -30,18 +30,13 @@ It can be used in applications requiring emotional analysis in English text.
|
|
| 30 |
|
| 31 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
| 32 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
| 33 |
-
| 0.
|
| 34 |
-
| 0.
|
| 35 |
-
| 0.
|
| 36 |
-
| 0.
|
| 37 |
-
|
| 38 |
-
- **Validation Accuracy**:
|
| 39 |
-
- Epoch 1: 0.9275
|
| 40 |
-
- Epoch 2: 0.9345
|
| 41 |
-
- Epoch 3: 0.940
|
| 42 |
-
- Epoch 4: 0.940
|
| 43 |
-
- **Test Accuracy**: 100% accuracy on the provided 10 test examples.
|
| 44 |
|
|
|
|
|
|
|
| 45 |
|
| 46 |
## Usage
|
| 47 |
```python
|
|
@@ -51,9 +46,9 @@ classifier = pipeline("text-classification", model="Zoopa/emotion-classification
|
|
| 51 |
text = "I am so happy today!"
|
| 52 |
result = classifier(text)
|
| 53 |
print(result)
|
| 54 |
-
|
| 55 |
|
| 56 |
## Limitations
|
| 57 |
- The model only supports English.
|
| 58 |
- The training dataset may contain biases, affecting model predictions on test data.
|
| 59 |
-
- Edge Cases like mixed emotions might reduce accuracy.
|
|
|
|
| 30 |
|
| 31 |
| Training Loss | Epoch | Step | Validation Loss | Accuracy |
|
| 32 |
|:-------------:|:-----:|:----:|:---------------:|:--------:|
|
| 33 |
+
| 0.5164 | 1.0 | 500 | 0.1887 | 0.9275 |
|
| 34 |
+
| 0.1464 | 2.0 | 1000 | 0.1487 | 0.9345 |
|
| 35 |
+
| 0.0994 | 3.0 | 1500 | 0.1389 | 0.9415 |
|
| 36 |
+
| 0.0701 | 4.0 | 2000 | 0.1479 | 0.94 |
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 37 |
|
| 38 |
+
- **Overall Training Loss**: 0.2081
|
| 39 |
+
- **Test Accuracy**: 100% accuracy on the provided 10 test examples.
|
| 40 |
|
| 41 |
## Usage
|
| 42 |
```python
|
|
|
|
| 46 |
text = "I am so happy today!"
|
| 47 |
result = classifier(text)
|
| 48 |
print(result)
|
| 49 |
+
```
|
| 50 |
|
| 51 |
## Limitations
|
| 52 |
- The model only supports English.
|
| 53 |
- The training dataset may contain biases, affecting model predictions on test data.
|
| 54 |
+
- Edge Cases like mixed emotions might reduce accuracy.
|