k折-交叉验证-展示过程

本文介绍了k折交叉验证的概念,并通过Python代码详细展示了如何定义和实现k折交叉验证的过程,包括获取每个折的数据和执行交叉验证的步骤。在代码中,通过get_k_fold函数划分数据,然后使用k_fold函数进行训练和验证,最后输出平均训练和验证的误差结果。
摘要由CSDN通过智能技术生成

先定义一个k交叉取训练集数据函数:

def get_k_fold(k,i,X,y):
    assert k> 1
    fold_size = X.shape[0]//k
    X_train,y_train = None,None
    for j in range(k):
        ix = slice(j * fold_size ,(j+1) * fold_size)
        #print('ix:',ix)
        X_part, y_part = X[ix, :], y[ix]      
        
        if j == i:
            print('[j==i]:j = %d,i = %d'%(j,i))
            X_valid,y_valid = X_part,y_part
        elif X_train is None:
            print('[X_train is None]:j = %d,i = %d'%(j,i))
            X_train ,y_train = X_part,y_part
        else:
            print('[X_train,X_part]:j = %d,i = %d'%(j,i))
            X_train = torch.cat((X_train,X_part),dim=0)            
            y_train = torch.cat((y_train,y_part),dim=0)
            print('X_train

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值