refactor: seperate train & sweep configs #9

twndus · May 16, 2024 · 771c0ca · 771c0ca
1 parent 9045b2f
commit 771c0ca
Show file tree

Hide file tree

Showing 8 changed files with 97 additions and 289 deletions.
diff --git a/_train.py b/_train.py
diff --git a/configs/_train_config.yaml b/configs/_train_config.yaml
diff --git a/configs/sweep_config.yaml b/configs/sweep_config.yaml
@@ -1,36 +1,5 @@
-# run config
-seed:       42 
-shuffle:    True
-model_dir:  outputs/models
-submit_dir: outputs/submissions
-data_dir:   data/
-log_dir:    logs/
-
-# wandb config
-wandb:      True # True/ False
-project:    YelpRecommendation
-notes:      "..."
-tags:       [test, yelp, cdae]
-
-# train config
-device:     cuda # cpu
-epochs:     1
-batch_size: 32
-lr:         0.001
-optimizer:  adamw
-loss:       bce 
-patience:   5
-top_n:      10
-
-# model config
-model_name: CDAE
-hidden_size:    64
-corruption_level: 0.6
-hidden_activation:  sigmoid
-output_activation:  sigmoid
-
-
 # sweep config
+## MF
 sweep_count: 3
 method:     random
 name:       sweep

diff --git a/configs/train_config.yaml b/configs/train_config.yaml
@@ -5,6 +5,7 @@ model_dir:  outputs/models
 submit_dir: outputs/submissions
 data_dir:   data/
 log_dir:    logs/
+sweep:      True 
 
 # wandb config
 wandb:      True # True/ False
@@ -17,20 +18,20 @@ device:     cuda # cpu
 epochs:     10
 batch_size: 32
 lr:         0.001
-optimizer:  sgd # adamw
-loss_name:  bpr # pointwise # bce 
+optimizer:  adam # adamw
+loss_name:  bce # bpr # pointwise # bce 
 patience:   5
 top_n:      10
 weight_decay: 0 #1e-5 
 
 # model config
-#model_name: CDAE 
-#negative_sampling: True # False
-#neg_times:  5   # this works only when negative_sampling == True, if value is 5, the number of negative samples will be 5 times the number of positive samples by users
-#hidden_size:    64
-#corruption_level: 0.6
-#hidden_activation:  sigmoid
-#output_activation:  sigmoid
+model_name: CDAE 
+negative_sampling: True # False
+neg_times:  5   # this works only when negative_sampling == True, if value is 5, the number of negative samples will be 5 times the number of positive samples by users
+hidden_size:    64
+corruption_level: 0.6
+hidden_activation:  sigmoid
+output_activation:  sigmoid
 
-model_name: MF 
-embed_size: 64
+#model_name: MF 
+#embed_size: 64
diff --git a/poetry.lock b/poetry.lock
diff --git a/pyproject.toml b/pyproject.toml
@@ -18,6 +18,7 @@ hydra-core = "^1.3.2"
 loguru = "^0.7.2"
 wandb = "^0.17.0"
 scikit-learn = "1.4.0"
+easydict = "^1.13"
 
 [tool.poetry.group.dev.dependencies]
 ipykernel = "^6.29.4"

diff --git a/sweep.py b/sweep.py