KBGAN/config_fb15k237.yaml at master · danecor/KBGAN · 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
pretrain_config: TransE
g_config: DistMult
d_config: TransE
test_batch_size: 100
task:
  dir: fb15k237
  type: link
adv:
  n_sample: 20
  n_epoch: 5000
  n_batch: 100
  epoch_per_test: 100
  temperature: 1.0
TransE:
  dim: 50
  margin: 3
  p: 1
  n_epoch: 1000
  n_batch: 100
  epoch_per_test: 50
  model_file: transe_backup.mdl
TransD:
  dim: 50
  margin: 3
  p: 1
  n_epoch: 1000
  n_batch: 100
  epoch_per_test: 50
  model_file: transd.mdl
DistMult:
  dim: 50
  lam: 1
  n_sample: 20
  sample_freq: 1
  n_epoch: 1000
  n_batch: 100
  epoch_per_test: 50
  model_file: distmult.mdl
ComplEx:
  dim: 25
  lam: 1
  n_sample: 20
  sample_freq: 1
  n_epoch: 1000
  n_batch: 100
  epoch_per_test: 50
  model_file: complex.mdl
log:
  dump_config: false
  to_file: false
  dir: ./log
  prefix: