diff --git a/configs/model/caum.yaml b/configs/model/caum.yaml index f936cda..9e4d361 100644 --- a/configs/model/caum.yaml +++ b/configs/model/caum.yaml @@ -47,12 +47,16 @@ outputs: "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 diff --git a/configs/model/cen_news_rec.yaml b/configs/model/cen_news_rec.yaml index aba8165..cf62ca6 100644 --- a/configs/model/cen_news_rec.yaml +++ b/configs/model/cen_news_rec.yaml @@ -41,12 +41,16 @@ outputs: "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 diff --git a/configs/model/dkn.yaml b/configs/model/dkn.yaml index e138b4f..11b2289 100644 --- a/configs/model/dkn.yaml +++ b/configs/model/dkn.yaml @@ -33,12 +33,16 @@ outputs: "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 diff --git a/configs/model/lstur.yaml b/configs/model/lstur.yaml index 47f7af4..c7e28f3 100644 --- a/configs/model/lstur.yaml +++ b/configs/model/lstur.yaml @@ -43,12 +43,16 @@ outputs: "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 diff --git a/configs/model/manner_cr_module.yaml b/configs/model/manner_cr_module.yaml index 83c4f8a..8a28358 100644 --- a/configs/model/manner_cr_module.yaml +++ b/configs/model/manner_cr_module.yaml @@ -34,12 +34,16 @@ outputs: "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 diff --git a/configs/model/manner_module.yaml b/configs/model/manner_module.yaml index 1b3adb8..be2c06a 100644 --- a/configs/model/manner_module.yaml +++ b/configs/model/manner_module.yaml @@ -18,12 +18,16 @@ outputs: "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: null scheduler: null diff --git a/configs/model/miner.yaml b/configs/model/miner.yaml index aeefaf4..8aa28de 100644 --- a/configs/model/miner.yaml +++ b/configs/model/miner.yaml @@ -39,12 +39,16 @@ outputs: "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 diff --git a/configs/model/mins.yaml b/configs/model/mins.yaml index 07aafaf..6514885 100644 --- a/configs/model/mins.yaml +++ b/configs/model/mins.yaml @@ -40,12 +40,16 @@ outputs: "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 diff --git a/configs/model/naml.yaml b/configs/model/naml.yaml index abf9aea..94a2483 100644 --- a/configs/model/naml.yaml +++ b/configs/model/naml.yaml @@ -38,12 +38,16 @@ outputs: "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 diff --git a/configs/model/npa.yaml b/configs/model/npa.yaml index 4b2b2c6..40282a3 100644 --- a/configs/model/npa.yaml +++ b/configs/model/npa.yaml @@ -32,12 +32,16 @@ outputs: "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 diff --git a/configs/model/nrms.yaml b/configs/model/nrms.yaml index b60ed12..13546b9 100644 --- a/configs/model/nrms.yaml +++ b/configs/model/nrms.yaml @@ -35,12 +35,16 @@ outputs: "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 diff --git a/configs/model/senti_debias.yaml b/configs/model/senti_debias.yaml index 8ff4251..ad7ca59 100644 --- a/configs/model/senti_debias.yaml +++ b/configs/model/senti_debias.yaml @@ -45,6 +45,8 @@ outputs: "target_sentiments", "hist_categories", "hist_sentiments", + "user_ids", + "cand_news_ids", ] # loss coefficients @@ -55,6 +57,8 @@ beta_coefficient: 10 top_k_list: [5, 10] num_categ_classes: 18 num_sent_classes: 3 +save_recs: False +recs_fpath: "${paths.output_dir}/recommendations.json" optimizer: null diff --git a/configs/model/sentirec.yaml b/configs/model/sentirec.yaml index def2ea3..8b9ade1 100644 --- a/configs/model/sentirec.yaml +++ b/configs/model/sentirec.yaml @@ -41,12 +41,16 @@ outputs: "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 diff --git a/configs/model/tanr.yaml b/configs/model/tanr.yaml index 18bfe2a..41765b3 100644 --- a/configs/model/tanr.yaml +++ b/configs/model/tanr.yaml @@ -40,12 +40,16 @@ outputs: "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