update dataset source

Catiks · Jul 29, 2024 · b206b60 · b206b60
1 parent 48de498
commit b206b60
Show file tree

Hide file tree

Showing 2 changed files with 7 additions and 3 deletions.
diff --git a/README.md b/README.md
@@ -27,14 +27,18 @@ pip install -r requirements.txt
 ```
 
 ### Dataset Preparation
-Get the beauty, toys-and-games, gowalla dataset under dataset folder
-Overall file structure:
+Get the [Amazon_Beauty](https://recbole.s3-accelerate.amazonaws.com/ProcessedDatasets/Amazon_ratings/Amazon_Beauty.zip), [Amazon_Toys_and_Games](https://recbole.s3-accelerate.amazonaws.com/ProcessedDatasets/Amazon_ratings/Amazon_Toys_and_Games.zip), [gowalla](https://drive.google.com/drive/folders/1vhRuvIzKYBSlaNFcD9Y3igWjBgaA) dataset under dataset folder
+
+Please modify the file name and arrange them in the following structure:
 ```
 LightGODE/
     ├─ Dataset/
         ├─ amazon-beauty
+            ├─ amazon-beauty.inter
         ├─ amazon-toys-games
+            ├─ amazon-toys-games.inter
         ├─ gowalla
+            ├─ gowalla.inter
 ```
 
 ### Running on different datasets:

diff --git a/run_recbole.py b/run_recbole.py
@@ -19,7 +19,7 @@
     parser.add_argument('--gpu', '-g', type=str, default='0', help='gpu id')
     parser.add_argument('--lr', '-l', type=float, default=0.001, help='learing rate')
     parser.add_argument('--weight_decay', '-w', type=float, default=1e-8, help='weight decay')
-    parser.add_argument('--tem', '-t', type=float, default=None, help='temporary parameters')
+    parser.add_argument('--tem', '-t', type=float, default=None, help='temperature')
     parser.add_argument('--suffix', '-s', type=str, default=None, help='suffix logger')
     parser.add_argument("--config_files", type=str, default=None, help="config files")
     parser.add_argument("--run", '-r', type=str, default=None, help="config files")