Add TransR

Author: jimmywangheng

transE_Bernoulli_pytorch.py

@@ -301,7 +301,7 @@ def __init__(self):
 				#if USE_CUDA:
 					#model.cuda()
 
-			# Check the result on validation set for every 5 epochs
+			# Evaluate on validation set for every 5 epochs
 			elif epoch % 5 == 0:
 				ent_embeddings = model.ent_embeddings.weight.data.cpu().numpy()
 				rel_embeddings = model.rel_embeddings.weight.data.cpu().numpy()

transH_Bernoulli_pytorch.py

@@ -312,7 +312,7 @@ def __init__(self):
 				#if USE_CUDA:
 					#model.cuda()
 
-			# Check the result on validation set for every 5 epochs
+			# Evaluate on validation set for every 5 epochs
 			elif epoch % 5 == 0:
 				ent_embeddings = model.ent_embeddings.weight.data
 				rel_embeddings = model.rel_embeddings.weight.data

transH_pytorch.py

@@ -304,7 +304,7 @@ def __init__(self):
 				#if USE_CUDA:
 					#model.cuda()
 
-			# Check the result on validation set for every 5 epochs
+			# Evaluate on validation set for every 5 epochs
 			elif epoch % 5 == 0:
 				ent_embeddings = model.ent_embeddings.weight.data
 				rel_embeddings = model.rel_embeddings.weight.data