Small changes

PatrickSVM · Jul 14, 2023 · 909d27b · 909d27b
1 parent b7442a9
commit 909d27b
Show file tree

Hide file tree

Showing 4 changed files with 2 additions and 4 deletions.
diff --git a/Master_Thesis_IKEA_final_handin.pdf → Thesis.pdf b/Master_Thesis_IKEA_final_handin.pdf → Thesis.pdf
diff --git a/recommenders/ikea/training/trainBidirGRU.py b/recommenders/ikea/training/trainBidirGRU.py
@@ -59,8 +59,7 @@ def train_BidirGRU4Rec(
     output_tokenizer=None,
 ):
     """
-    Trains GRU4Rec model with provided arguments.
-    Also used to pretrain embedding layer for diversity rewards
+    Trains Bidirectional-GRU4Rec model with provided arguments.
     """
 
     disable_pro_bar = not progress_bar

diff --git a/recommenders/ikea/training/trainGRU.py b/recommenders/ikea/training/trainGRU.py
@@ -59,7 +59,6 @@ def train_GRU4Rec(
 ):
     """
     Trains GRU4Rec model with provided arguments.
-    Also used to pretrain embedding layer for diversity rewards
     """
 
     disable_pro_bar = not progress_bar

diff --git a/recommenders/ikea/training/trainSMORL.py b/recommenders/ikea/training/trainSMORL.py
@@ -62,7 +62,7 @@ def train_SMORL(
     save_pretrained_embeddings=True,
 ):
     """
-    Trains smorl model with provided arguments.
+    Trains SMORL model with provided arguments.
     """
 
     disable_pro_bar = not progress_bar
-Original file line number
+Diff line change
@@ Expand Up / @@ -59,7 +59,6 @@ def train_GRU4Rec( @@
     ):
         """
         Trains GRU4Rec model with provided arguments.
-        Also used to pretrain embedding layer for diversity rewards
         """
         disable_pro_bar = not progress_bar
@@ Expand Down @@