Fix TransE config
@@ -5,8 +5,7 @@ job:
model: transe
dataset:
- name: fb15k-237
-# name: toy
+ name: toy
train:
batch_size: 256
@@ -19,8 +18,8 @@ train:
type: negative_sampling
negative_sampling:
- num_negatives_o: 1000
- num_negatives_s: 1000
+ num_negatives_o: 3
+ num_negatives_s: 3
sampling_type: uniform
valid:
@@ -35,7 +34,7 @@ eval:
transe:
class_name: TransE
entity_embedder:
- dim: 512
+ dim: 128
initialize: uniform_
initialize_args:
uniform_ :
@@ -47,7 +46,7 @@ transe:
weight: 1.e-05
weighted: False
relation_embedder:
- dim: 512
+ dim: 128
initialize: uniform_
initialize_args:
uniform_ :
