목록code tip (1)
rueki
imbalance data일때 weightedRandomSampler 적용하기
이미지 분류에서 클래스마다 존재하는 데이터의 수가 항상 비슷하게 존재하지 않은다. 불균형적인 데이터일때 적용할 수 있는 방법 중 하나를 소개한다. WeightedRandomSampler라는 방법인데 Dataloader를 불러올때 같이 선언할 수 있다고 한다. import torch from torch.utils.data import Dataset, DataLoader, WeightedRandomSampler ############################################### #위에 데이터셋 class, 및 등등 선언되어있다 가정# ############################################### #class 0 : 43200개, class 1 : 4800개 class..
code tip
2021. 8. 30. 22:14