DataLoader 썸네일형 리스트형 [pytorch] Dataset & Dataloader Dataset 클래스 하나의 데이터를 어떻게 가져올 것인가? 데이터 입력 형태를 정의하는 클래스 데이터를 입력하는 방식을 표준화 이미지, 텍스트, 오디오 등 데이터 포맷에 따라 다른 입력을 정의해야 한다. import torch from torch.utils.data import Dataset, DataLoader class CustomDataset(Dataset): def __init__(self, text, labels): # 초기 데이터 생성 방법 지정 self.labels = labels self.data = text def __len__(self): # 데이터의 전체 길이 return len(self.labels) def __getitem__(self, idx): # index값을 주었을 때 반.. 더보기 이전 1 다음
