torch로 dataset 만들기

PyTorch에서 나만의 Dataset을 만들 때 반드시 작성해야 하는 함수는 다음과 같다.__init__: 데이터셋의 초기화를 담당__len__: 데이터셋의 크기(길이)를 반환__getitem__: 주어진 인덱스에 해당하는 데이터를 반환PyTorch의 Dataset 클래스를 상속받아야 한다.아래는 그 예시import torchfrom torch.utils.data import Dataset, DataLoaderclass MyDataset(Dataset): def __init__(self, data, labels): """ 데이터셋의 초기화를 수행합니다. Args: data (list or numpy array): 데이터 리스트 또는 배열 ..

2024.07.07