sklearn.model_selection.KFold中random_state

使用K折交叉验证来反应模型的平均性能,首先应该保证每次数据集的划分是一样的,即第一个模型的训练集应该与第二个模型的训练集应该相同,这样得出来的对比结果才更有说服力。那么下面就是介绍如何使得输入到不同的模型时保证数据集是一样的。
令shuffle=True和random_state=某个整数

from sklearn.model_selection import KFold
import numpy as np
X = np.arange(24).reshape(12,2)
kf = KFold(n_splits=5,shuffle=True,random_state=1)
for train_index , test_index in kf.split(X):
	print(f'train_index:{train_index} , test_index: {test_index} ')

运行几次,实验结果均如下所示,所以,划分数据集在每次送到不同的模型时,每个fold的结果是一样的。
在这里插入图片描述

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

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

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

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值