Update model configs with hyperparameters for saving recommendations …

…lists
andreeaiana · Mar 19, 2024 · 1b33d38 · 1b33d38
1 parent eca9f54
commit 1b33d38
Show file tree

Hide file tree

Showing 14 changed files with 56 additions and 0 deletions.
diff --git 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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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
@@ -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