File size: 788 Bytes
9c58361 |
1 2 3 4 5 6 7 8 9 10 11 12 13 14 15 16 17 18 19 20 21 22 23 24 25 26 27 28 29 30 31 32 33 34 35 36 37 38 39 40 41 42 43 44 45 46 47 |
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
|