ashnadua01 commited on
Commit
bd35adb
1 Parent(s): 7bf364d

update model card README.md

Browse files
Files changed (1) hide show
  1. README.md +10 -10
README.md CHANGED
@@ -19,11 +19,11 @@ should probably proofread and complete it, then remove this comment. -->
19
 
20
  This model is a fine-tuned version of [ai4bharat/indic-bert](https://huggingface.co/ai4bharat/indic-bert) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
- - Loss: 0.1079
23
- - Precision: 0.9141
24
- - Recall: 0.9076
25
- - F1: 0.9109
26
- - Accuracy: 0.9652
27
 
28
  ## Model description
29
 
@@ -43,8 +43,8 @@ More information needed
43
 
44
  The following hyperparameters were used during training:
45
  - learning_rate: 2e-05
46
- - train_batch_size: 20
47
- - eval_batch_size: 20
48
  - seed: 42
49
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
  - lr_scheduler_type: linear
@@ -54,9 +54,9 @@ The following hyperparameters were used during training:
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
56
  |:-------------:|:-----:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
57
- | 0.1584 | 1.0 | 8806 | 0.1524 | 0.8780 | 0.8698 | 0.8739 | 0.9499 |
58
- | 0.1149 | 2.0 | 17612 | 0.1217 | 0.9032 | 0.8961 | 0.8996 | 0.9606 |
59
- | 0.0925 | 3.0 | 26418 | 0.1079 | 0.9141 | 0.9076 | 0.9109 | 0.9652 |
60
 
61
 
62
  ### Framework versions
 
19
 
20
  This model is a fine-tuned version of [ai4bharat/indic-bert](https://huggingface.co/ai4bharat/indic-bert) on an unknown dataset.
21
  It achieves the following results on the evaluation set:
22
+ - Loss: 0.0999
23
+ - Precision: 0.9203
24
+ - Recall: 0.9141
25
+ - F1: 0.9172
26
+ - Accuracy: 0.9672
27
 
28
  ## Model description
29
 
 
43
 
44
  The following hyperparameters were used during training:
45
  - learning_rate: 2e-05
46
+ - train_batch_size: 10
47
+ - eval_batch_size: 10
48
  - seed: 42
49
  - optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
50
  - lr_scheduler_type: linear
 
54
 
55
  | Training Loss | Epoch | Step | Validation Loss | Precision | Recall | F1 | Accuracy |
56
  |:-------------:|:-----:|:-----:|:---------------:|:---------:|:------:|:------:|:--------:|
57
+ | 0.1412 | 1.0 | 17611 | 0.1358 | 0.8961 | 0.8834 | 0.8897 | 0.9555 |
58
+ | 0.1024 | 2.0 | 35222 | 0.1076 | 0.9133 | 0.9076 | 0.9104 | 0.9641 |
59
+ | 0.076 | 3.0 | 52833 | 0.0999 | 0.9203 | 0.9141 | 0.9172 | 0.9672 |
60
 
61
 
62
  ### Framework versions