파이토치 3.7 커스텀 데이터셋(Custom Dataset)

torch.utils.data.Dataset을 상속받아 직접 커스텀 데이터셋을 만드는 경우가 있다. class CustomDataset(torch.utils.data.Dataset): def __init__(self): 데이터셋의 전처리를 해주는 부분 def __len__(self): 데이터셋의 길이. 즉, 총 샘플의 수를 적어주는 부분 def __getitem__(self, idx): 데이터셋에서 특정 1개의 샘플을 가져오는 함수 https://wikidocs.net/57165 03-07 커스텀 데이터셋(Custom Dataset) 앞 내용을 잠깐 복습해봅시다. 파이토치에서는 데이터셋을 좀 더 쉽게 다룰 수 있도록 유용한 도구로서 torch.utils.data.Dataset과 torch.utils.data.… wikidocs.net...

원문링크 : 파이토치 3.7 커스텀 데이터셋(Custom Dataset)