K-fold交叉验证(python版)

最新推荐文章于 2025-04-10 12:23:41 发布

supergxt

最新推荐文章于 2025-04-10 12:23:41 发布

阅读量2k

点赞数

CC 4.0 BY-SA版权

分类专栏：机器学习

本文链接：https://blog.csdn.net/supergxt/article/details/109428211

本文介绍了如何使用Python中的sklearn库进行k-fold交叉验证，包括其原理和原创版本的实现，旨在帮助理解训练集和验证集的划分方法。

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

k-fold交叉验证

sklearn可以实现训练集交叉验证划分训练集和验证集
这里提供一个原创版本。

class My_cv_iterator(): # 自写交叉验证
    def __init__(self,data,label,cv_num):
        '''
        data ：输入特征
        label : 输入标签
        cv_num : k-fold 折数

        return
        返回k-fold的iterator 包括step, traindata, validdata

        e.g
        for step, train, valid in My_cv_iterator()
        '''
        self.data = data
        self.label = label
        self.cv_num = cv_num
        train = np.hstack