Create README.md
Browse files
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
|