-
Notifications
You must be signed in to change notification settings - Fork 77
/
window_8tracks_sgnn.yml
36 lines (32 loc) · 1.06 KB
/
window_8tracks_sgnn.yml
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
---
type: window # single|window, maybe add opt
key: stan #added to the csv names
evaluation: evaluation #evaluation|evaluation_last|evaluation_multiple
data:
name: 8tracks #added in the end of the csv names
folder: data/8tracks/slices/
prefix: playlists-8tracks
slices: 5 #only window
#opts: {sessions_test: 10}
results:
folder: results/window/8tracks/
#pickle_models: results/models/music-window/ # not working for tensorflow models
metrics:
- class: accuracy.HitRate
length: [3,5,10,15,20]
- class: accuracy.MRR
length: [3,5,10,15,20]
- class: accuracy_multiple.NDCG
length: [3,5,10,15,20]
- class: coverage.Coverage
length: [20]
- class: popularity.Popularity
length: [20]
- class: saver.Saver
length: [50]
- class: time_memory_usage.Time_usage_training
- class: time_memory_usage.Time_usage_testing
#- class: time_memory_usage.Memory_usage
algorithms:
- class: sgnn.gnn.GGNN
params: { hidden_size: 100, out_size: 100, step: 1, nonhybrid: True, batch_size: 100, epoch_n: 10, batch_predict: True, lr: 0.002, l2: 5.00E-05, lr_dc: 0.46, lr_dc_step: 7}