Update README.md
Browse files
README.md
CHANGED
@@ -11,7 +11,7 @@ tags:
|
|
11 |
- playstore
|
12 |
- multi-class
|
13 |
- classification
|
14 |
-
|
15 |
inference: true
|
16 |
---
|
17 |
|
@@ -26,7 +26,7 @@ Trained on 9000 samples of English App Descriptions and associated categories of
|
|
26 |
|
27 |
## Fine-tuning
|
28 |
|
29 |
-
The model was fine-tuned for 5 epochs with a batch size of 16, a learning rate of 2e-05, and a maximum sequence length of 512. Since this was a classification task, the model was trained with a cross-entropy loss function. The best evaluation f1 score achieved by the model was 0.9034534096919489, found after 4 epochs. The accuracy of the model on the test set was
|
30 |
|
31 |
## How to use
|
32 |
|
|
|
11 |
- playstore
|
12 |
- multi-class
|
13 |
- classification
|
14 |
+
license: "mit"
|
15 |
inference: true
|
16 |
---
|
17 |
|
|
|
26 |
|
27 |
## Fine-tuning
|
28 |
|
29 |
+
The model was fine-tuned for 5 epochs with a batch size of 16, a learning rate of 2e-05, and a maximum sequence length of 512. Since this was a classification task, the model was trained with a cross-entropy loss function. The best evaluation f1 score achieved by the model was 0.9034534096919489, found after 4 epochs. The accuracy of the model on the test set was 0.9033.
|
30 |
|
31 |
## How to use
|
32 |
|