dataset:
  name: wnrr
  num_entities: 40943
  num_relations: 11
  test_size: 3134
  train_size: 86835
  valid_size: 3034
entity_ranking:
  chunk_size: 5000
  metrics_per:
    relation_type: true
eval:
  batch_size: 256
import:
- rotate
- reciprocal_relations_model
job:
  device: cuda:1
lookup_embedder:
  dim: 256
  initialize: xavier_normal_
  initialize_args:
    normal_:
      mean: 0.0
      std: 2.0006032717029774e-05
    uniform_:
      a: -0.9279830623610272
    xavier_normal_:
      gain: 1.0
    xavier_uniform_:
      gain: 1.0
model: reciprocal_relations_model
negative_sampling:
  implementation: triple
  num_samples:
    o: 291
    s: 18
reciprocal_relations_model:
  base_model:
    relation_embedder:
      dim: 128
    type: rotate
rotate:
  entity_embedder:
    dropout: -0.09153870348946103
    regularize_weight: 1.0054555427885155e-06
  l_norm: 2.0
  relation_embedder:
    dropout: -0.18838447788070495
    regularize_weight: 1.5712131774007265e-17
train:
  auto_correct: true
  batch_size: 128
  loss: bce
  loss_arg: 9.809548087591716
  lr_scheduler: ReduceLROnPlateau
  lr_scheduler_args:
    factor: 0.95
    mode: max
    patience: 6
    threshold: 0.0001
  max_epochs: 400
  optimizer:
    default:
      args:
        lr: 0.20986162552270587
  subbatch_auto_tune: true
  type: negative_sampling
valid:
  early_stopping:
    patience: 10
    threshold:
      epochs: 50
      metric_value: 0.05

