KvsAll:
  label_smoothing: 0.0
  query_types:
    _po: true
    s_o: false
    sp_: true
ax_search:
  num_sobol_trials: -1
  num_trials: 10
  parameter_constraints: []
  parameters: []
  sobol_seed: 0
conve:
  2D_aspect_ratio: 2
  class_name: ConvE
  convolution_bias: true
  entity_embedder:
    +++: +++
    dropout: 0.2
    type: lookup_embedder
  feature_map_dropout: 0.2
  filter_size: 3
  padding: 0
  projection_dropout: 0.3
  relation_embedder:
    +++: +++
    dropout: 0.2
    type: lookup_embedder
  round_dim: false
  stride: 1
dataset:
  +++: +++
  files:
    +++: +++
    entity_ids:
      filename: entity_ids.del
      type: map
    entity_strings:
      filename: entity_ids.del
      type: map
    relation_ids:
      filename: relation_ids.del
      type: map
    relation_strings:
      filename: relation_ids.del
      type: map
    test:
      filename: test.del
      type: triples
    train:
      filename: train.del
      type: triples
    valid:
      filename: valid.del
      type: triples
  name: fb15k-237
  num_entities: -1
  num_relations: -1
  pickle: true
entity_ranking:
  chunk_size: -1
  filter_splits:
  - train
  - valid
  filter_with_test: true
  hits_at_k_s:
  - 1
  - 3
  - 10
  - 50
  - 100
  - 200
  - 300
  - 400
  - 500
  - 1000
  metrics_per:
    argument_frequency: false
    head_and_tail: false
    relation_type: false
  tie_handling: rounded_mean_rank
eval:
  batch_size: 100
  num_workers: 0
  pin_memory: false
  split: valid
  trace_level: epoch
  type: entity_ranking
grid_search:
  parameters:
    +++: +++
  run: true
import:
- reciprocal_relations_model
- transformer
job:
  device: cuda:0
  type: train
lookup_embedder:
  class_name: LookupEmbedder
  dim: 100
  dropout: 0.0
  initialize: normal_
  initialize_args:
    +++: +++
  normalize:
    p: -1.0
  pretrain:
    ensure_all: false
    model_filename: ''
  regularize: lp
  regularize_args:
    +++: +++
    p: 2
    weighted: false
  regularize_weight: 0.0
  round_dim_to: []
  sparse: false
manual_search:
  configurations: []
  run: true
model: reciprocal_relations_model
negative_sampling:
  filtering:
    implementation: fast_if_available
    o: false
    p: false
    s: false
    split: ''
  frequency:
    smoothing: 1
  implementation: triple
  num_samples:
    o: -1
    p: 0
    s: 3
  sampling_type: uniform
  shared: false
  shared_type: default
  with_replacement: true
random_seed:
  default: -1
  numba: -1
  numpy: -1
  python: -1
  torch: -1
reciprocal_relations_model:
  base_model:
    +++: +++
    type: transformer
  class_name: ReciprocalRelationsModel
search:
  device_pool: []
  num_workers: 1
  on_error: abort
  type: ax
train:
  abort_on_nan: true
  auto_correct: false
  batch_size: 512
  checkpoint:
    every: 5
    keep: 3
  loss: kl
  loss_arg: .nan
  lr_scheduler: ReduceLROnPlateau
  lr_scheduler_args:
    +++: +++
    factor: 0.5
    mode: max
    patience: 2
  lr_warmup: 50
  max_epochs: 500
  num_workers: 0
  optimizer: Adam
  optimizer_args:
    +++: +++
    lr: 0.001
  pin_memory: false
  split: train
  subbatch_auto_tune: false
  subbatch_size: -1
  trace_level: epoch
  type: 1vsAll
  visualize_graph: false
transformer:
  class_name: Transformer
  entity_embedder:
    +++: +++
    dim: 320
    dropout: 0.6
    initialize: xavier_normal_
    type: lookup_embedder
  initialize: normal_
  initialize_args:
    +++: +++
    mean: 0.0
    std: 0.02
  relation_embedder:
    +++: +++
    dim: 320
    dropout: 0.6
    initialize: xavier_normal_
    type: lookup_embedder
user:
  +++: +++
valid:
  early_stopping:
    min_threshold:
      epochs: 0
      metric_value: 0.0
    patience: 5
  every: 5
  metric: mean_reciprocal_rank_filtered_with_test
  metric_expr: float("nan")
  split: valid
  trace_level: epoch
verbose: true
