Movie Recommendation 대회 log - 221220

Custom Dataset,

Pytorch Custom Dataset 구성

  • 3가지가 필수이다
    1. __init__
    2. __len__ : 데이터셋의 길이 즉 샘플의 수를 리턴
    3. __getitem__ : 특정 1개의 샘플을 인덱스에 맞게 가져오는 함수

https://wikidocs.net/57165


© 2023. All rights reserved.