K-fold交叉验证(python版)

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