-
Notifications
You must be signed in to change notification settings - Fork 7
/
test.yaml
43 lines (39 loc) · 1006 Bytes
/
test.yaml
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
save_sst_embed: True
# pretrain_model_file_path: 'saved/saved_FairGo_GCN/FairGo_GCN-ml-1M-pretrain.pth'
LABEL_FIELD: label
threshold: {'rating': 3.0}
TIME_FIELD: timestamp
sst_attr_list: ['gender']
load_col:
inter: [user_id,item_id,rating,timestamp]
user: [user_id,gender]
model: FairGo_GCN
aggr_method: 'LBA'
vs_weights: [4,1]
embedding_size: 64
n_layers: 2
dis_hidden_size_list: [16,8,4]
filter_hidden_size_list: [128,64]
activation: leakyrelu
fair_weight: 0.1
# gcn
gcn_n_layers: 2
hidden_channels: 32
gcn_dropout: 0.2
gcn_act: relu
# training settings
pretrain_epochs: 600
train_epoch_interval: 5
weight_decay: 0.0001
# evalution settings
eval_args:
split: {'RS':[8,1,1]}
group_by: user
order: RO
mode: uni100
metrics: ["NDCG","Recall","Hit","MRR","DifferentialFairness","GiniIndex","PopularityPercentage",
"ValueUnfairness","AbsoluteUnfairness","UnderUnfairness","OverUnfairness","NonParityUnfairness"]
valid_metric: NDCG@5
topk: [5]
popularity_ratio: 0.1
eval_batch_size: 8192