Я просматривал коды онлайн , и мне интересно, что делает следующий код?
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. цель составляет таблицу каких-то показателей. Но подумал, понимая остальные коды. это не легко понять.