卜若的代碼筆記-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則---------