Changed hyperparameters in RotatE

src/models/train_model.py

@@ -23,19 +23,19 @@ def get_model(name: str, triples, inverse_triples, regularizer):
     }
     lr_dict = {
         "TransE": 0.01,
-        "RotatE": 5.0e-6,
+        "RotatE": 0.01,
         "ConvE": 0.001,
         "DistMult": 0.1,
     }
     epochs_dict = {
         "TransE": 500,
-        "RotatE": 6000,
+        "RotatE": 600,
         "ConvE": 500,
         "DistMult": 300,
     }
     bs_dict = {
         "TransE": 1440,
-        "RotatE": 100000,
+        "RotatE": 1440,
         "ConvE": 128,
         "DistMult": 4000,
     }