New hyperparameters

src/models/train_model.py

@@ -12,13 +12,13 @@ from src.models.predict_model import evaluate
 
 def get_model(name: str, triples, inverse_triples, regularizer):
     model_dict = {
-        "TransE": TransE(triples_factory=triples, embedding_dim=20,
+        "TransE": TransE(triples_factory=triples, embedding_dim=200,
                          scoring_fct_norm=1, regularizer=regularizer),
         "RotatE": RotatE(triples_factory=triples, embedding_dim=250, regularizer=regularizer),
         "ConvE": ConvE(triples_factory=inverse_triples, embedding_dim=200,
                        feature_map_dropout= 0.2,  input_dropout= 0.2,
                        output_dropout=0.3),
-        "DistMult": DistMult(triples_factory=triples, embedding_dim=100,
+        "DistMult": DistMult(triples_factory=triples, embedding_dim=200,
                              regularizer=regularizer),
     }
     lr_dict = {
@@ -72,7 +72,7 @@ if __name__ == '__main__':
         model_results[name] = {}
         print(name)
 
-        regularizer = LpRegularizer(weight=0.001, p=2.0)
+        regularizer = LpRegularizer(weight=0.0001, p=2.0)
         model, lr, epochs, batch_size = get_model(name, training_triples_factory,
                                                   training_inverse, regularizer)