def METHOD_NAME(self, test_entity, test_rel):
    # Q1884806	P2102	ff4f077428e5023035faa1cd7971e5afae17b707
    rel_embedding = pd.read_csv(os.path.join(DATA_DIR, self.dataset_dir, 'rel_embedding.tsv'), sep='\t',
                                header=None)
    ent_embedding = pd.read_csv(os.path.join(DATA_DIR, self.dataset_dir, 'ent_embedding.tsv'), sep='\t',
                                header=None)
    attr_embedding = pd.read_csv(os.path.join(DATA_DIR, self.dataset_dir, 'attr_embedding.tsv'), sep='\t',
                                 header=None)
    bias_embedding = pd.read_csv(os.path.join(DATA_DIR, self.dataset_dir, 'bias_embedding.tsv'), sep='\t',
                                 header=None)
    relation_idx = self.rel2idx[test_rel]
    rel = torch.tensor(rel_embedding.iloc[relation_idx].values).to(self.device)
    attr = torch.tensor(attr_embedding.iloc[relation_idx].values).to(self.device)
    bias = torch.tensor(bias_embedding.iloc[relation_idx].values).to(self.device)
    head_idx = self.entity2idx[test_entity]
    heads_idx = torch.Tensor([head_idx])
    tails_idx = torch.Tensor(self.subgraph_extractor.extract_neighbour_from_idx(head_idx))
    head = torch.tensor(ent_embedding.iloc[heads_idx].values).to(self.device)
    tails = torch.tensor(ent_embedding.iloc[tails_idx].values).to(self.device)
    repeat_num = len(tails)
    heads = head.repeat(repeat_num, 1).to(self.device)
    rels = rel.repeat(repeat_num, 1).to(self.device)
    attrs = attr.repeat(1, 1).to(self.device)
    numerical_prediction, pred_idx = self.score_model.get_numerical_prediction(head=[head_idx], attr=attrs,
                                                                               bias=bias)
    relation_idx = [relation_idx] * len(heads)
    scores = self.score_model.triple_distance_transR(heads, tails, rels, relation_idx)
    _, indices_top_k = torch.topk(scores, k=len(scores), largest=self.largest)
    labels_top_k = [self.idx2entity[tails_idx[index].item()] for index in indices_top_k]
    return labels_top_k[0], numerical_prediction