-
Notifications
You must be signed in to change notification settings - Fork 43
/
NeuRec.properties
52 lines (39 loc) · 978 Bytes
/
NeuRec.properties
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
[default]
######## model
recommender=SGL
# model configuration directory
config_dir=./conf
gpu_id=0
gpu_mem=0.5
######## dataset
data.input.path=dataset
data.input.dataset=gowalla
# data.column.format = UIRT, UIT, UIR, UI
data.column.format=UI
# separator "\t" " ","::", ","
data.convert.separator=','
######## pre-processing/filtering
user_min=0
item_min=0
######## data splitting
# splitter = ratio, loo, given
splitter=given
# train set ratio if splitter=ratio
ratio=0.8
by_time=False
######## evaluating
# metric = Precision, Recall, MAP, NDCG, MRR
metric=["Precision", "Recall", "NDCG", "MAP", "MRR"]
# topk is int or list of int
topk=[20]
# group_view is list or None, e.g. [10, 20, 30, 40]
group_view=None
rec.evaluate.neg=0
test_batch_size=128
num_thread=8
start_testing_epoch=0
# data pre-process
# binThold = -1.0 do nothing
# binThold = value, rating > value is changed to 1.0 other is 0.0.
# data.convert.binarize.threshold=0
#will be used to evaluate.