boostcampaitech7 / level1-imageclassification-cv-16

level1-imageclassification-cv-16 created by GitHub Classroom
3 stars 1 forks source link

Feat/ 원핫인코딩 #20

Closed kkh090 closed 1 month ago

kkh090 commented 1 month ago

Fixes #6 Fixes#4 merge 이후 처리하길 바람

PR 타입(하나 이상의 PR 타입을 선택해주세요)

반영 브랜치

kkh090/issue6 -> Development

변경 사항

CustomDataset에 Onehot encoding을 받을 수 있도록 코드 추가

테스트 결과

CustomDataset()의 매개변수 중 one_hot=True를 넘겨주면 label값을 one_hot으로 받을 수 있음