base:
  data:
    groups:
      filepath_or_buffer:
          - 'datasets/stratigis/groupsWithHighRatings5.txt'
          - 'datasets/stratigis/groupsWithModerateRatings5untested.txt'
    testdata:
      filepath_or_buffer: 'datasets/fake_data.csv'
      sep: '\t'
      skiprows: 1
      names: [ 'userId', 'itemId', 'rating', 'timestamp']
    test:
      filepath_or_buffer: 'datasets/stratigis/ratings.csv'
      sep: ','
      skiprows: 1
      names: [ 'userId', 'itemId', 'rating', 'timestamp']
    ml32m:
      filepath_or_buffer: 'datasets/ml-32m/ratings.csv'
      sep: ','
      skiprows: 1
      names: [ 'userId', 'itemId', 'rating', 'timestamp']
    ml100k:
        filepath_or_buffer: 'datasets/ml-100k/u.data'
        sep: '\t'
        skiprows: 0
        names: [ 'userId', 'itemId', 'rating', 'timestamp']
    ml1m:
      filepath_or_buffer: 'datasets/ml-1m/ratings.dat'
      sep: '::'
      names: [ 'userId', 'itemId', 'rating', 'timestamp' ]
    tags:
        tags_file: 'datasets/stratigis/tags.csv'
  model:
    gmf:
      learning_rate: 0.005
      weight_decay: 0.0000001
      latent_dim: 8
      epochs: 30
      num_negative: 10
      batch_size: 1024
      cuda: False
      optimizer_name: 'adam'
    mlp:
      learning_rate: 0.005
      weight_decay: 0.0000001
      latent_dim: 8
      epochs: 30
      num_negative: 10
      batch_size: 1024
      cuda: False
      optimizer_name: 'adam'
    als:
      learning_rate: 0.1
      latent_dim: 100
      epochs: 10
      reg_term: 0.001

    bpr:
      learning_rate: 0.01
      latent_dim: 100
      epochs: 10
      reg_term: 0.001
    emf:
      learning_rate: 0.01
      reg_term: 0.001
      expl_reg_term: 0.0
      latent_dim: 80
      epochs: 10
      positive_threshold: 3
      knn: 10
    mf:
      learning_rate: 0.01
      reg_term: 0.001
      expl_reg_term: 0.0
      latent_dim: 80
      epochs: 10
      positive_threshold: 3
      knn: 10
    autoencoder:
      learning_rate: 0.005
      weight_decay: 0.0000001
      hidden_layer_features: 8
      epochs: 30
      cuda: False
      optimizer_name: 'adam'
      positive_threshold: 3
      knn: 10
      expl: true
  explainer:
    lore4groups:
      n_similar_for_tree: 100
      rating_threshold_for_like: 3.0
      max_tree_depth: 5
      top_n_labels: 5000
      min_rating_for_history: 1.0
      similarity_threshold: 0.1