def get_test_loader(self):
test_dataset = LSTMSeq2SeqDataset(self.x_test_path, self.y_test_path,
self.ko_voc, self.en_voc,
self.seq_len)
# dataset을 인자로 받아 data를 뽑아냄
test_loader = DataLoader(test_dataset, batch_size=self.batch_size)
return test_loader
def get_val_loader(self):
# 재현을 위해 랜덤시드 고정
seed_val = 42
torch.manual_seed(seed_val)
# path를 불러와서 train_loader를 만드는 함수
val_dataset = LSTMSeq2SeqDataset(self.x_val_path, self.y_val_path, self.ko_voc, self.en_voc,
self.args.sequence_size)
point_sampler = torch.utils.data.RandomSampler(val_dataset) # data의 index를 반환하는 함수, suffle를 위한 함수
# dataset을 인자로 받아 data를 뽑아냄
val_loader = DataLoader(val_dataset, batch_size=self.args.batch_size, sampler=point_sampler)
return val_loader