不平衡数据集采样器Imbalanced Dataset Sampler使用自定义数据集时报错
Imbalanced Dataset Sampler是一个针对不均衡数据集的重采样器。但直接使用自定义数据集时会报错。查GitHub发现Imbalanced Dataset Sampler需要调用数据集的get_labels() 用于返回你数据集所有的标签值，而自定义的数据集一般是没有这个方法的。所以根据自己的数据集定义一个输出自己全部标签的方法就可以了，同时使用Sampler就不能在datalo
·
Imbalanced Dataset Sampler是一个针对不均衡数据集的重采样器。但直接使用自定义数据集时会报错。
查GitHub发现Imbalanced Dataset Sampler需要调用数据集的get_labels() 用于返回你数据集所有的标签值，而自定义的数据集一般是没有这个方法的。
所以在自己的数据集类里定义一个输出自己全部标签的方法就可以了，同时使用Sampler就不能在dataloader中使用shuffle了
魔乐社区（Modelers.cn) 是一个中立、公益的人工智能社区，提供人工智能工具、模型、数据的托管、展示与应用协同服务，为人工智能开发及爱好者搭建开放的学习交流平台。社区通过理事会方式运作，由全产业链共同建设、共同运营、共同享有，推动国产AI生态繁荣发展。
更多推荐
0
0- 0
已为社区贡献2条内容
所有评论(0)