深度学习的分类任务数据划分方式

[jinxianwei]

分类任务的深度学习模型，在将数据划分为训练集和测试集时，使用的是torch.utils.data的random_split方法，该方法没有考虑类别不均衡现象，当数据样本类别极度不均衡时，训练集和测试集的数据分布会非常不同，由训练集训练的模型，就不可能在测试集上表现良好，需要修改。