nsi319 commited on
Commit
d5640e0
1 Parent(s): 6d0dcc5

Update README.md

Browse files
Files changed (1) hide show
  1. README.md +2 -2
README.md CHANGED
@@ -11,7 +11,7 @@ tags:
11
  - playstore
12
  - multi-class
13
  - classification
14
- liscence: "mit"
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 90.33.
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