[Pytorch] Custom Dataset

참고 자료
https://wikidocs.net/53560
유원준 님의 Pytorch로 시작하는 딥러닝 입문 책 자료를 참고하였습니다. 

 

Pytorch에서는 데이터셋을 상속받아 직접 데이터셋을 지정하는 식으로 많이 사용한다고 한다. 

일반적으로 데이터셋을 구축할 때는 다음 세가지를 정의한다고 한다.

 

__init__(self) : 데이터셋의 전처리를 해주는 부분

 

__len__(self) : 데이터셋의 길이, 샘플 수

 

__getitem__(self,idx) : 데이터셋에서 특정 1개의 샘플 가져오기

 

앞서 구현하였던 다중 선형 회귀를 돌리는 데이터셋을 정의하는 커스텀 데이터셋을 다음과 같이 사용할 수 있다.