Notice
Recent Posts
Recent Comments
«   2024/11   »
1 2
3 4 5 6 7 8 9
10 11 12 13 14 15 16
17 18 19 20 21 22 23
24 25 26 27 28 29 30
Tags
more
Archives
Today
Total
관리 메뉴

No Limitation

[Pytorch] Custom Dataset 본문

ML & DL & RL

[Pytorch] Custom Dataset

yesungcho 2022. 2. 3. 19:32

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

 

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

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

 

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

 

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

 

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

 

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

 

 

 

'ML & DL & RL' 카테고리의 다른 글

[Pytorch] 다중 클래스 분류 회귀  (0) 2022.02.04
[Pytorch] 로지스틱 회귀분석  (0) 2022.02.04
[Pytorch] Mini Batch Size와 Data Loader  (0) 2022.02.03
[Pytorch] Class를 이용한 Model 구현  (0) 2022.02.03
[Pytorch] nn.Module  (0) 2022.02.03