-
Notifications
You must be signed in to change notification settings - Fork 3
/
Copy pathcsbm_lp.yaml
72 lines (64 loc) · 1.11 KB
/
csbm_lp.yaml
1
2
3
4
5
6
7
8
9
10
11
12
13
14
15
16
17
18
19
20
21
22
23
24
25
26
27
28
29
30
31
32
33
34
35
36
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
seml:
name: trn_lp
executable: exp_train.py
project_root_dir: ../../
output_dir: log/training
slurm:
experiments_per_job: 1
sbatch_options:
gres: gpu:1 # num GPUs
mem: 10G # memory
cpus-per-task: 4 # num cores
time: 0-00:30 # max time, D-HH:MM
fixed:
train_params:
inductive: True
model_params:
label: LP
model: None
use_label_propagation: True
lp_layers: 50
lp_use_clamping: False
other_params:
sacred_metrics: False
grid:
data_params:
K:
type: choice
options:
- 0.1
#- 0.5 Different K don't matter
#- 1
#- 1.5
#- 2
#- 5
model_params:
lp_alpha:
type: choice
options:
- 0.00
- 0.05
- 0.10
- 0.15
- 0.20
- 0.25
- 0.30
- 0.35
- 0.40
- 0.45
- 0.50
- 0.55
- 0.60
- 0.65
- 0.70
- 0.75
- 0.80
- 0.85
- 0.90
- 0.95
- 1.00
seed:
type: range
min: 0
max: 30
step: 1