KHGNN/prod_config.yaml at main · iMoonLab/KHGNN · GitHub

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

data:
  #name: cora
  #name: pubmed
  #name: citeseer
  #name: news20
  #name: dblp4k-conf
  #name: dblp4k-paper
  #name: coauthorship_dblp
  #name: imdb4k
  name: dblp4k-term
  #name: cooking200

  num_train: 20
  num_val: 100
  ft_noise_level: 0.0
  self_loop: True
  random_split: True
  test_ind_ratio: 0.2
model:

  #name: gcn
  #name: gat
  #name: hgnn
  name: hgnnp
  #name: kerhgnn
  #name: unignn
  #name: unigat
  #name: hnhn


  use_norm: True
  p_min: -0.5
  p_max: 0.5
  mu: 1
  hid: 32
  num_layer: 2
  kernel_type: poly
optim:
  lr: 0.1
  lr_p: 0.001

wd: ${hydra:runtime.cwd}
data_marker: ${data.name}__${data.num_train}-${data.num_val}__noise-${data.ft_noise_level}
model_marker: ${model.name}-${model.p_min}-${model.p_max}
task: ${data_marker}__${model_marker}
res_path: ${wd}/cache/prod/${task}

hydra:
  run:
    dir: ${res_path}/${now:%Y-%m-%d}_${now:%H-%M-%S}
  sweep:
    dir: ${res_path}/${now:%Y-%m-%d}_${now:%H-%M-%S}
    subdir: ${hydra.job.num}