StratifiedKFold
最新推荐文章于 2022-11-12 18:08:30 发布
StratifiedKFold是一种交叉验证方法,即使在设置shuffle=True时,也能保持每个类别内部的比例。它在随机划分数据集时,确保了每个fold中各类别的比例与原始数据集相同。尽管如此,返回的索引序列仍然有序,方便后续训练时进行数据洗牌。这种方法在处理不平衡数据集时尤其重要,保证了验证集的多样性。
摘要由CSDN通过智能技术生成