十折交叉验证python_Python实现K折交叉验证法的方法步骤

学习器在测试集上的误差我们通常称作“泛化误差”。要想得到“泛化误差”首先得将数据集划分为训练集和测试集。那么怎么划分呢？常用的方法有两种，k折交叉验证法和自助法。介绍这两种方法的资料有很多。下面是k折交叉验证法的python实现。

##一个简单的2折交叉验证

from sklearn.model_selection import KFold

import numpy as np

X=np.array([[1,2],[3,4],[1,3],[3,5]])

Y=np.array([1,2,3,4])

KF=KFold(n_splits=2) #建立4折交叉验证方法 查一下KFold函数的参数

for train_index,test_index in KF.split(X):

print("TRAIN:",train_index,"TEST:",test_index)

X_train,X_test=X[train_index],X[test_index]

Y_train,Y_test=Y[train_index],Y[test_index]

print(X_train,X_test)

print(Y_train,Y_test)

#小结：KFold这个包 划分k折交叉验证的时候，是以TEST集的顺序为主的，举例来说，如果划分4折交叉验证，那么TEST选取的顺序为[0].[1],[2],[3]。

#提升

import numpy as np

from sklearn.model_selection import KFold

#Sample=np.random.rand(50,15) #建立一个50行12列的随机数组