-
Notifications
You must be signed in to change notification settings - Fork 8
/
miner.yaml
58 lines (51 loc) · 1.17 KB
/
miner.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
44
45
46
47
48
49
50
51
52
53
54
55
56
57
58
_target_: newsreclib.models.general_rec.miner_module.MINERModule
# training strategy
dual_loss_training: False
dual_loss_coef: null
loss: "cross_entropy_loss"
late_fusion: False
temperature: None
# news encoder
dataset_attributes: ${data.dataset_attributes}
attributes2encode: ["title"]
# text encoder
use_plm: True
plm_model: null
frozen_layers: null
apply_reduce_dim: True
text_embed_dim: 768
news_embed_dim: 256
dropout_probability: 0.2
use_categ_bias: True
pretrained_categ_embeddings_path: null
num_context_codes: 32
context_code_dim: 200
score_type: "weighted" # choose between weighted, mean, max
# outputs
outputs:
train: ["preds", "targets", "cand_news_size"]
val: ["preds", "targets", "cand_news_size"]
test:
[
"preds",
"targets",
"cand_news_size",
"hist_news_size",
"target_categories",
"target_sentiments",
"hist_categories",
"hist_sentiments",
"user_ids",
"cand_news_ids",
]
# evaluation
top_k_list: [5, 10]
num_categ_classes: 18
num_sent_classes: 3
save_recs: False
recs_fpath: "${paths.output_dir}/recommendations.json"
optimizer:
_target_: torch.optim.Adam
_partial_: true
lr: 0.0001
scheduler: null