Merge pull request #42 from boostcampaitech5/feat-#14/LightGCNtrans

Feat #14/lgcnLSTMattn

DKT/config/config_lgcntrans.json

@@ -0,0 +1,79 @@
+{
+    "name": "lgcnLSTMattn",
+    "n_gpu": 1,
+
+    "arch": {
+        "type": "lgcnLSTMattn",
+        "args": {
+            "user_num": 7442,
+            "item_num": 9454,
+            "embedding_dim": 64,
+            "gamma": 1e-4,
+            "lambda": 0.8
+        }
+    },
+    "data_loader": {
+        "type": "lgcnLSTMattnDataLoader",
+        "args":{
+            "data_dir": "/opt/ml/input/data/",
+            "batch_size": 512,
+            "shuffle": true,
+            "num_workers": 2,
+            "validation_split": 0.2
+        }
+    },
+    "optimizer": {
+        "type": "Adam",
+        "args":{
+            "lr": 0.001,
+            "weight_decay": 0,
+            "amsgrad": true
+        }
+    },
+    "loss": "lgcnLSTMattn_loss",
+    "metrics": [
+        "accuracy",
+        "auc"
+    ],
+    "lr_scheduler": {
+        "type": "StepLR",
+        "args": {
+            "step_size": 50,
+            "gamma": 0.1
+        }
+    },
+    "model": {
+        "max_seq_len": 200,
+        "hidden_dim": 256,
+        "n_layers": 2,
+        "n_heads": 4,
+        "drop_out": 0.4,
+        "gcn_n_layes": 2,
+        "alpha": 1.0,
+        "beta": 1.0
+    },
+    "trainer": {
+        "n_epochs": 60,
+        "batch_size": 70,
+        "lr": 0.000001,
+        "clip_grad" : 10,
+        "patience": 100,
+        "log_step": 50,
+
+        "save_dir": "saved/",
+        "save_period": 1,
+        "verbosity": 2,
+
+        "monitor": "min val_loss",
+        "early_stop": 10,
+
+        "tensorboard": false
+    },
+    "test": {
+        "data_dir": "~/input/data/test_data_modify.csv",
+        "model_dir": "./saved/models/LGCNtrans/0518_033541/model_best.pth",
+        "submission_dir": "~/level2_dkt-recsys-09/DKT/submission/lgcnLSTMattn_submission.csv",
+        "sample_submission_dir": "~/input/data/sample_submission.csv",
+        "batch_size": 512
+    }
+}