Add TransH

Author: jimmywangheng

transE_Bernoulli_pytorch.py

@@ -115,8 +115,8 @@ def __init__(self):
 	validTotal, validList, validDict = loadTriple('./data/' + args.dataset, 'valid2id.txt')
 	tripleTotal, tripleList, tripleDict = loadTriple('./data/' + args.dataset, 'triple2id.txt')
 	with open(os.path.join('./data/', args.dataset, 'head_tail_proportion.pkl'), 'rb') as fr:
-	tail_per_head = pickle.load(fr)
-	head_per_tail = pickle.load(fr)
+		tail_per_head = pickle.load(fr)
+		head_per_tail = pickle.load(fr)
 
 	config = Config()
 	config.dataset = args.dataset
@@ -317,7 +317,7 @@ def __init__(self):
 					torch.save(model, os.path.join('./model/' + args.dataset, filename))
 				else:
 					meanrank_not_decrease_time += 1
-					# If the result hasn't improved for consecutive 5 epochs, decrease learning rate
+					# If the result hasn't improved for consecutive 5 evaluations, decrease learning rate
 					if meanrank_not_decrease_time == 5:
 						lr_decrease_time += 1
 						if lr_decrease_time == config.early_stopping_round: