我需要在PyTorch中实现一个多标签图像分类模型。但是我的数据不平衡,所以我使用PyTorch中的WeightedRandomSampler创建了一个自定义数据加载器。但是当我遍历定制的dataloader时,我得到了错误:IndexError: list index out of range 使用此链接实现了以下代码:https://discuss.pytorch</
我有一个不平衡的数据集,并希望对类进行欠采样,即overrepresented.How do I go。我想使用称重随机采样器,但我也对其他建议持开放态度。到目前为止,我假设我的代码必须像下面这样的结构。但我不知道该怎么做。trainset = datasets.ImageFolder(path_train,transform=transform) ... sampler = data.WeightedRandomSampler