谈一谈二分类比赛中常用的KFold, StratifiedKFold K折交叉切分

一颗西蓝花

于 2019-03-07 11:27:37 发布

阅读量3.1k

点赞数 4

分类专栏： Python

本文链接：https://blog.youkuaiyun.com/weixin_41160054/article/details/88293148

版权

本文介绍了二分类比赛中KFold和StratifiedKFold的区别，StratifiedKFold保证了分层采样，维持样本类别比例。同时，讨论了KFold与enumerate的联合使用场景，常用于赛事中的数据划分。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

我在最近的好几场二分类赛事中，看到别人分享的kernel，都用到了KFold，因此我准备详细记录一下KFold和StratifiedKFold的用法。

1. KFold 和StratifiedKFold有什么区别

import numpy as np
from sklearn.model_selection import KFold,StratifiedKFold

X = np.array([
    [1,2,3,4],
    [11,12,13,14],
    [21,22,23,24],