разъяснение метода: def get_batch (self, er_vocab, er_vocab_pairs, idx): - PullRequest
0 голосов
/ 23 апреля 2020

Я просматривал коды онлайн , и мне интересно, что делает следующий код?

def get_batch(self, er_vocab, er_vocab_pairs, idx):
    batch = er_vocab_pairs[idx:idx+self.batch_size]
    targets = np.zeros((len(batch), len(d.entities)))
    for idx, pair in enumerate(batch):
        targets[idx, er_vocab[pair]] = 1.
    targets = torch.FloatTensor(targets)
    if self.cuda:
        targets = targets.cuda()
    return np.array(batch), targets

Итак, насколько я понимаю, batch = er_vocab_pairs[idx:idx+self.batch_size] получают подмножество пакетное количество элементов из er_vocab_pairs. цель составляет таблицу каких-то показателей. Но подумал, понимая остальные коды. это не легко понять.

Добро пожаловать на сайт PullRequest, где вы можете задавать вопросы и получать ответы от других членов сообщества.
...