VSRAJHAVEL commited on
Commit
dee181e
·
verified ·
1 Parent(s): d058e08

Upload folder using huggingface_hub

Browse files
Files changed (3) hide show
  1. configs/bbbp.yaml +23 -0
  2. configs/esol.yaml +23 -0
  3. configs/tox21.yaml +23 -0
configs/bbbp.yaml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset: bbbp
2
+ task_type: classification
3
+ num_tasks: 1
4
+ metric: roc_auc
5
+ higher_is_better: true
6
+
7
+ training:
8
+ batch_size: 32
9
+ epochs: 50
10
+ lr: 0.001
11
+ weight_decay: 0.0001
12
+ patience: 15
13
+ grad_clip: 1.0
14
+ lr_scheduler: reduce_on_plateau
15
+ lr_patience: 5
16
+ lr_factor: 0.5
17
+ seeds: [42, 123, 999]
18
+
19
+ model:
20
+ hidden_dim: 64
21
+ num_layers: 3
22
+ dropout: 0.2
23
+ residual: true
configs/esol.yaml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset: esol
2
+ task_type: regression
3
+ num_tasks: 1
4
+ metric: rmse
5
+ higher_is_better: false
6
+
7
+ training:
8
+ batch_size: 32
9
+ epochs: 50
10
+ lr: 0.001
11
+ weight_decay: 0.0001
12
+ patience: 15
13
+ grad_clip: 1.0
14
+ lr_scheduler: reduce_on_plateau
15
+ lr_patience: 5
16
+ lr_factor: 0.5
17
+ seeds: [42, 123, 999]
18
+
19
+ model:
20
+ hidden_dim: 64
21
+ num_layers: 3
22
+ dropout: 0.2
23
+ residual: true
configs/tox21.yaml ADDED
@@ -0,0 +1,23 @@
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
 
1
+ dataset: tox21
2
+ task_type: multitask_classification
3
+ num_tasks: 12
4
+ metric: roc_auc
5
+ higher_is_better: true
6
+
7
+ training:
8
+ batch_size: 64
9
+ epochs: 50
10
+ lr: 0.001
11
+ weight_decay: 0.0001
12
+ patience: 15
13
+ grad_clip: 1.0
14
+ lr_scheduler: reduce_on_plateau
15
+ lr_patience: 5
16
+ lr_factor: 0.5
17
+ seeds: [42, 123, 999]
18
+
19
+ model:
20
+ hidden_dim: 64
21
+ num_layers: 3
22
+ dropout: 0.2
23
+ residual: true