Update README.md
Browse files
README.md
CHANGED
|
@@ -1,3 +1,12 @@
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
|
| 1 |
T5-small is a fine-tuned version of DeBERTa model. The dataset used for fine tuning is the IMDB dataset from Hugging Face. The number of instances used for training is 12 500, and 2 500 for testing. Metrics used for evaluation are classification oriented, because the model is used for sentiment analysis. The number of epochs used for training is 5, and the learning rate was set to lr = 5e-5. The results obtained are:
|
| 2 |
Accuracy: 0.9312
|
| 3 |
Precision: 0.9268
|
|
|
|
| 1 |
+
---
|
| 2 |
+
datasets:
|
| 3 |
+
- imdb
|
| 4 |
+
language:
|
| 5 |
+
- en
|
| 6 |
+
metrics:
|
| 7 |
+
- accuracy
|
| 8 |
+
library_name: transformers
|
| 9 |
+
---
|
| 10 |
T5-small is a fine-tuned version of DeBERTa model. The dataset used for fine tuning is the IMDB dataset from Hugging Face. The number of instances used for training is 12 500, and 2 500 for testing. Metrics used for evaluation are classification oriented, because the model is used for sentiment analysis. The number of epochs used for training is 5, and the learning rate was set to lr = 5e-5. The results obtained are:
|
| 11 |
Accuracy: 0.9312
|
| 12 |
Precision: 0.9268
|