manandey commited on
Commit
2c54c8e
·
1 Parent(s): d0c9d63

Create README.md

Browse files
Files changed (1) hide show
  1. README.md +48 -0
README.md ADDED
@@ -0,0 +1,48 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ ---
2
+ license: mit
3
+ datasets:
4
+ - taskydata/tasky_or_not
5
+ language:
6
+ - en
7
+ metrics:
8
+ - f1
9
+ - accuracy
10
+ - recall
11
+ - precision
12
+ pipeline_tag: text-classification
13
+ ---
14
+
15
+ **Hyperparameters:**
16
+
17
+ - learning rate: 2e-5
18
+ - weight decay: 0.01
19
+ - per_device_train_batch_size: 8
20
+ - per_device_eval_batch_size: 8
21
+ - gradient_accumulation_steps:1
22
+ - eval steps: 50000
23
+ - max_length: 512
24
+ - num_epochs: 1
25
+ - hidden_dropout_prob: 0.3
26
+ - attention_probs_dropout_prob: 0.25
27
+
28
+ **Dataset version:**
29
+ - tasky_or_not/10xp3nirstbbflanseuni_10xc4
30
+
31
+ **Checkpoint:**
32
+
33
+ - 300000 steps
34
+
35
+ **Results on Validation set:**
36
+
37
+ | **Step** | **Training Loss** | **Validation Loss** | **Accuracy** | **Precision** | **Recall** | **F1** |
38
+ |:--------:|:-----------------:|:-------------------:|:------------:|:-------------:|:----------:|:--------:|
39
+ | 50000 | 0.020800 | 0.192550 | 0.970363 | 0.990686 | 0.949654 | 0.969736 |
40
+ | 100000 | 0.015200 | 0.264168 | 0.969427 | 0.994374 | 0.944196 | 0.968636 |
41
+ | 150000 | 0.012900 | 0.146541 | 0.981440 | 0.994599 | 0.968138 | 0.981190 |
42
+ | 200000 | 0.011100 | 0.319310 | 0.970516 | 0.998871 | 0.942097 | 0.969654 |
43
+ | 250000 | 0.008000 | 0.204103 | 0.976309 | 0.996226 | 0.956241 | 0.975824 |
44
+ | 300000 | 0.006100 | 0.096262 | 0.988053 | 0.994676 | 0.981358 | 0.987972 |
45
+ | 350000 | 0.005800 | 0.162989 | 0.983663 | 0.994730 | 0.972478 | 0.983478 |
46
+
47
+ **Wandb logs:**
48
+ - https://wandb.ai/manandey/taskydata/runs/y3j3fbkh?workspace=user-manandey