hype_kg/codes/dataloader.py [20:35]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(self, triples, nentity, nrelation, negative_sample_size, train_ans, mode):
        assert mode == 'tail-batch'
        self.len = len(triples)
        self.triples = triples
        self.nentity = nentity
        self.nrelation = nrelation
        self.negative_sample_size = negative_sample_size
        self.mode = mode
        self.count = self.count_frequency(triples, train_ans)
        self.true_tail = train_ans
        self.qtype = self.triples[0][-1]

    def __len__(self):
        return self.len
    
    def __getitem__(self, idx):
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -



hype_kg/codes/dataloader.py [91:106]:
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -
    def __init__(self, triples, nentity, nrelation, negative_sample_size, train_ans, mode):
        assert mode == 'tail-batch'
        self.len = len(triples)
        self.triples = triples
        self.nentity = nentity
        self.nrelation = nrelation
        self.negative_sample_size = negative_sample_size
        self.mode = mode
        self.count = self.count_frequency(triples, train_ans)
        self.true_tail = train_ans
        self.qtype = self.triples[0][-1]
        
    def __len__(self):
        return self.len
    
    def __getitem__(self, idx):
- - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - - -