TensorDataset은 Dataset을 상속한 Class이며 입력데이터 X와 레이블 Y를 묶어 놓은 컨테이너라고 생각하자.
이 TensorDataset을 DataLoader에 전달하면 이후의 학습 for loop에서 데이터의 일부만을 선택적으로 추출할 수 있다.
(이 때, TensorDataset에는 텐서형태만 전달이 가능함)
처음사용할때는 굳이 이것을 사용해야할까라는 의구심이 샘솟을 수 있으나, 익숙해지면 그 소중함을 느낄 수 있다.
'Deep Learning > PyTorch' 카테고리의 다른 글
| 나만의 신경망 모듈 만들어보기! (0) | 2020.07.14 |
|---|---|
| Overfitting 문제 해결해보기 (0) | 2020.07.14 |
| Multi Layer Perceptron (다층 퍼셉트론?) 만들어보기 (0) | 2020.07.14 |
| 다중 분류를 위한 Logistic Regression Model 만들기 (0) | 2020.07.14 |
| Logistic Regression 구현해보기 (0) | 2020.07.14 |
