Update README.md
Browse files
README.md
CHANGED
@@ -74,14 +74,14 @@ model-index:
|
|
74 |
|
75 |
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on a subset of the [amazon-polarity dataset](https://huggingface.co/datasets/amazon_polarity).
|
76 |
|
77 |
-
<b>[Update 10/10/23]</b> The model has been retrained with an improvement on the loss, f1 score and accuracy. It achieves the following results on the evaluation set:
|
78 |
- Loss: 0.116
|
79 |
- Accuracy: 0.961
|
80 |
- F1_score: 0.960
|
81 |
|
82 |
## Model description
|
83 |
|
84 |
-
This sentiment classifier has been trained on
|
85 |
|
86 |
## Intended uses & limitations
|
87 |
```python
|
|
|
74 |
|
75 |
This model is a fine-tuned version of [distilbert-base-uncased](https://huggingface.co/distilbert-base-uncased) on a subset of the [amazon-polarity dataset](https://huggingface.co/datasets/amazon_polarity).
|
76 |
|
77 |
+
<b>[Update 10/10/23]</b> The model has been retrained on a larger part of the dataset with an improvement on the loss, f1 score and accuracy. It achieves the following results on the evaluation set:
|
78 |
- Loss: 0.116
|
79 |
- Accuracy: 0.961
|
80 |
- F1_score: 0.960
|
81 |
|
82 |
## Model description
|
83 |
|
84 |
+
This sentiment classifier has been trained on 360_000 samples for the training set, 40_000 samples for the validation set and 40_000 samples for the test set.
|
85 |
|
86 |
## Intended uses & limitations
|
87 |
```python
|