상세 컨텐츠

본문 제목

[Pytorch] Dataset , Dataloader 뜻과 사용법

AI beginner by 비전공자

by Semi_AI_Expert 2022. 10. 12. 16:41

본문

Dataset , Dataloader :학습 모델에 맞게 학습데이터 모음과 테스트데이터 모음을 변경해주는 역할

 

1. torch.utils.data.Dataset

  사용가능 데이터 : pytorch library에 있는 data set & 미리 준비된 data set 도 사용가능

     csv파일의 경우 :  Dataset class 를 상속받아 구현 가능

관련글 더보기