in src/models.py [0:0]
def get_queries(self, queries, target='rhs'):
lhs = self.entity(queries[:, 0]).data
rel = self.relation(queries[:, 1]).data
rhs = self.entity(queries[:, 2]).data
if target == 'rhs':
queries = lhs + rel
elif target == 'lhs':
queries = -1 * (rel - rhs)
if target == 'rel':
queries = -1 * (lhs - rhs)
return queries