시작하기무료로 시작하기

PyTorch 데이터셋

PyTorch 데이터셋 지식을 다시 한 번 되살려 볼까요?

모델 학습을 시작하려면 데이터를 올바른 형식으로 로드해 모델에 전달해야 해요. PyTorch에서는 이 작업을 Dataset과 DataLoader가 담당합니다. 우선 수질 적합성(water potability) 데이터를 위한 PyTorch Dataset을 만들어 보겠습니다.

이번 연습에서는 CSV 파일에서 데이터를 로드하는 WaterDataset 클래스를 정의할 거예요. 이를 위해 PyTorch의 Dataset이 가져야 하는 세 가지 메서드를 구현해야 합니다:

  • 데이터를 로드하는 .__init__()
  • 데이터 크기를 반환하는 .__len__()
  • 단일 샘플의 특징과 레이블을 추출하는 .__getitem()__

필요한 임포트는 이미 준비되어 있습니다:

import pandas as pd
from torch.utils.data import Dataset

이 연습은 강의의 일부입니다

PyTorch로 배우는 Intermediate Deep Learning

강의 보기

실습형 인터랙티브 연습

이 예제를 이 샘플 코드를 완성하여 풀어보세요.

class WaterDataset(Dataset):
    def __init__(self, csv_path):
        super().__init__()
        # Load data to pandas DataFrame
        df = ____
        # Convert data to a NumPy array and assign to self.data
        ____ = ____.____
코드 편집 및 실행