-
Notifications
You must be signed in to change notification settings - Fork 0
/
Copy pathtest_douban-movie.yml
86 lines (78 loc) · 1.68 KB
/
test_douban-movie.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
37
38
39
40
41
42
43
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
59
60
61
62
63
64
65
66
67
68
69
70
71
72
73
74
75
76
77
78
79
80
81
82
83
84
85
86
experiment:
dataset:
name: douban-movie
item:
path: datasets/douban-movie/processed/item.csv
extra_features: [name_EN, name_CN]
user:
path: datasets/douban-movie/processed/user.csv
extra_features: [name]
ratings:
path: datasets/douban-movie/processed/rating.csv
timestamp: False
preprocess:
# - method: filter_by_rating
# parameters:
# threshold: 20
# - method: binarize
# parameters:
# threshold: 4
- method: filter_kcore
parameters:
k: 20
iterations: 1
target: user # user or rating
split:
seed: 42
# test:
# method: random_by_ratio
# level: global
# p: 0.2
# validation:
# method: random_by_ratio
# level: global
# p: 0.2
# test:
# method: timestamp_by_ratio
# level: user
# p: 0.1
# validation:
# level: user
# method: timestamp_by_ratio
# p: 0.2
# test:
# method: fixed_timestamp
# # type: global_level
# timestamp: 890000000
# validation:
# method: fixed_timestamp
# timestamp: 880000000
test:
method: k_fold
k: 2
level: "user"
models:
- name: deepwalk_based
config:
parameters:
walk_len: 10
p: 1.0
q: 1.0
n_walks: 50
embedding_size: 64
epochs: 1
- name: deepwalk_based
config:
parameters:
walk_len: 10
p: 0.8
q: 0.6
n_walks: 50
embedding_size: 64
epochs: 1
evaluation:
k: 5
relevance_threshold: 0
metrics: [MAP, nDCG]
report:
file: "experiment_results/douban-movie/run1.csv"