AdamCodd commited on
Commit
2bd5a95
1 Parent(s): 0dd4594

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
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 180_000 samples for the training set, 20_000 samples for the validation set and 20_000 samples for the test set.
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