卜若的代码笔记-sklearn-第三章：knn（3）-对数据进行十则交叉划分

1.主要使用sklearn的KFolder工具

from sklearn.model_selection import KFold
import numpy as np

data = np.array([['王大'], ['王二'], ['王三'], ['王四'],['王五'],['王六'],['王七'],['王八'],['王九'],['王十']])
label = np.array([1, 2, 3, 4,5,6,7,8,9,10])
kf = KFold(n_splits=10)
num = kf.get_n_splits(data)


index = 0
for train_index, test_index in kf.split(data,label):
    train_data, train_label = data[train_index], label[train_index]
    test_data,test_label = data[test_index],label[test_index]
    print("x_train = \n{0}\nx_label = \n{1}\n----训练^----\n".format(train_data,train_label))
    print("x_test = \n{0}\nx_label = \n{1}\n----测试^----\n".format(test_data, test_label))

    print("\n\n----------第{0}则---------\n\n".format(index))
    index = index+1
    pass

演示结果：

x_train = 
[['王二']
 ['王三']
 ['王四']
 ['王五']
 ['王六']
 ['王七']
 ['王八']
 ['王九']
 ['王十']]
x_label = 
[ 2  3  4  5  6  7  8  9 10]
----训练^----

x_test = 
[['王大']]
x_label = 
[1]
----测试^----



----------第0则---------


x_train = 
[['王大']
 ['王三']
 ['王四']
 ['王五']
 ['王六']
 ['王七']
 ['王八']
 ['王九']
 ['王十']]
x_label = 
[ 1  3  4  5  6  7  8  9 10]
----训练^----

x_test = 
[['王二']]
x_label = 
[2]
----测试^----



----------第1则---------


x_train = 
[['王大']
 ['王二']
 ['王四']
 ['王五']
 ['王六']
 ['王七']
 ['王八']
 ['王九']
 ['王十']]
x_label = 
[ 1  2  4  5  6  7  8  9 10]
----训练^----

x_test = 
[['王三']]
x_label = 
[3]
----测试^----



----------第2则---------


x_train = 
[['王大']
 ['王二']
 ['王三']
 ['王五']
 ['王六']
 ['王七']
 ['王八']
 ['王九']
 ['王十']]
x_label = 
[ 1  2  3  5  6  7  8  9 10]
----训练^----

x_test = 
[['王四']]
x_label = 
[4]
----测试^----



----------第3则---------


x_train = 
[['王大']
 ['王二']
 ['王三']
 ['王四']
 ['王六']
 ['王七']
 ['王八']
 ['王九']
 ['王十']]
x_label = 
[ 1  2  3  4  6  7  8  9 10]
----训练^----

x_test = 
[['王五']]
x_label = 
[5]
----测试^----



----------第4则---------


x_train = 
[['王大']
 ['王二']
 ['王三']
 ['王四']
 ['王五']
 ['王七']
 ['王八']
 ['王九']
 ['王十']]
x_label = 
[ 1  2  3  4  5  7  8  9 10]
----训练^----

x_test = 
[['王六']]
x_label = 
[6]
----测试^----



----------第5则---------


x_train = 
[['王大']
 ['王二']
 ['王三']
 ['王四']
 ['王五']
 ['王六']
 ['王八']
 ['王九']
 ['王十']]
x_label = 
[ 1  2  3  4  5  6  8  9 10]
----训练^----

x_test = 
[['王七']]
x_label = 
[7]
----测试^----



----------第6则---------


x_train = 
[['王大']
 ['王二']
 ['王三']
 ['王四']
 ['王五']
 ['王六']
 ['王七']
 ['王九']
 ['王十']]
x_label = 
[ 1  2  3  4  5  6  7  9 10]
----训练^----

x_test = 
[['王八']]
x_label = 
[8]
----测试^----



----------第7则---------


x_train = 
[['王大']
 ['王二']
 ['王三']
 ['王四']
 ['王五']
 ['王六']
 ['王七']
 ['王八']
 ['王十']]
x_label = 
[ 1  2  3  4  5  6  7  8 10]
----训练^----

x_test = 
[['王九']]
x_label = 
[9]
----测试^----



----------第8则---------


x_train = 
[['王大']
 ['王二']
 ['王三']
 ['王四']
 ['王五']
 ['王六']
 ['王七']
 ['王八']
 ['王九']]
x_label = 
[1 2 3 4 5 6 7 8 9]
----训练^----

x_test = 
[['王十']]
x_label = 
[10]
----测试^----



----------第9则---------