dgcf_amazon-book.yml

experiment:
  backend: pytorch
  data_config:
    strategy: fixed
    train_path: ../data/{0}/train.tsv
    test_path: ../data/{0}/test.tsv
  dataset: amazon-book
  top_k: 20
  evaluation:
    cutoffs: [20]
    simple_metrics: [Recall, nDCG]
  gpu: 0
  external_models_path: ../external/models/__init__.py
  models:
    external.DGCF:
      meta:
        hyper_opt_alg: grid
        verbose: True
        save_weights: False
        validation_rate: 1
        validation_metric: Recall@20
        restore: False
        write_best_iterations: True
      lr: 0.001
      epochs: 3000
      factors: 64
      batch_size: 4096
      l_w_bpr: 1e-3
      l_w_ind: 0.01
      n_layers: 1
      routing_iterations: 2
      intents: 4
      seed: 42
      early_stopping:
        patience: 40
        mode: auto
        monitor: Recall@20
        verbose: True