config_sample.yml

models:
  rnn:
    embedding_dim: [256]
    batch: 64
    epochs: [50]
    model: ['LSTM']
    window: [3]
    bi: [False]
    num_units: [512]
  music2vec:
    window: [5]
    epochs: [5]
    down_sample: [1e-3]
    learning_rate: [0.025]
    embedding_dim: [300]
    negative_sample: [20]
  doc2vec:
    window: [10]
    epochs: [10]
    down_sample: [1e-4]
    learning_rate: [0.025]
    embedding_dim: [50]
    negative_sample: [10]
  glove:
    window: [10]
    embedding_dim: [100]
    epochs: [15]
    learning_rate: [0.025]
session:
  interval: 30
evaluation:
  dataset: 'sample'
  cross-validation: 5
  k: 5
  topN: 5
results:
  full: 'outputs/sample.csv'
embeddings:
  music2vec: 
    usage: True
  doc2vec: 
    usage: False
  glove: 
    usage: False
  rnn: 
    usage: False