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
____ = ____.____