交叉验证之留一交叉验证python代码

从csv文件中读取数据,由于样本量只有30,因此选择采用留一交叉验证方法,实验了一圈,发现一些问题在网上没有找到很好地解决办法,特记录如下,方便其他人使用。

主要代码如下:使用LeaveOneOut()将数据的索引分成测试和训练两部分,

注意:这里LeaveOneOut()返回的是数据的行索引,并不是数据本身,因此采用iloc方法根据索引读取到数据。

from sklearn.model_selection import LeaveOneOut
loo = LeaveOneOut()
for train_index, test_index in loo.split(data):#data为读取进来的csv文件数据
    x_train = pd.DataFrame(data.iloc[train_index, 1:])
    x_test = pd.DataFrame(data.iloc[test_index, 1:])
    y_train = pd.DataFrame(data.iloc[train_index, 0:1])
    y_test = pd.DataFrame(data.iloc[test_index, 0:1])

 

  • 5
    点赞
  • 23
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值