-
Notifications
You must be signed in to change notification settings - Fork 51
/
Copy pathbasic_configuration_v030.yml
56 lines (56 loc) · 1.53 KB
/
basic_configuration_v030.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
experiment:
version: 0.3.1
dataset: cat_dbpedia_movielens_1m
data_config:
strategy: dataset
dataset_path: ../data/movielens_1m/dataset.tsv
side_information:
- dataloader: ChainedKG
map: ../data/cat_dbpedia_movielens_1m/map.tsv
features: ../data/cat_dbpedia_movielens_1m/features.tsv
properties: ../data/cat_dbpedia_movielens_1m/properties.conf
# prefiltering:
# strategy: user_average # Not applied in the paper experiments
splitting:
save_on_disk: True
save_folder: ../data/cat_dbpedia_movielens_1m/splitting/
test_splitting:
strategy: temporal_hold_out
test_ratio: 0.2
validation_splitting:
strategy: temporal_hold_out
test_ratio: 0.2
top_k: 50
evaluation:
cutoffs: [10, 5]
simple_metrics: [nDCG,Precision,ItemCoverage,EPC,Gini]
relevance_threshold: 1
gpu: 1
external_models_path: ../external/models/__init__.py
models:
Random:
meta:
verbose: True
save_recs: True
seed: 42
external.MostPop:
meta:
verbose: True
save_recs: True
validation_metric: nDCG@10
ItemKNN:
meta:
verbose: True
save_recs: True
validation_metric: nDCG@10
neighbors: [50, 70, 100]
similarity: [cosine, euclidean]
implementation: standard
AttributeItemKNN:
meta:
verbose: True
save_recs: True
validation_metric: nDCG@10
loader: ChainedKG
neighbors: [50, 70, 100]
similarity: [braycurtis, manhattan]