前言

为了 更好的训练数据并且更好测试模型，一般做机器学习之前都会进行训练集和测试集的切分。

train_test_split实现

其实我们可以先把数据的输入X和输出向量y进行一个水平拼接，然后随机之后拆开，但是过程比较麻烦。在sklearn中shuffle的并不是训练集，而是训练集长度大小的随机索引。

产生随机索引值

shuffle_indexes=np.random.permutation(len(X)) #X为数据的输入
shuffle_indexes

np.random.permutation( x )这个函数可以产生x和随机数，并且随机数的范围是0～x

shuffle_indexes=np.random.permutation(3)
shuffle_indexes

array([2, 0, 1])

设置切分的比例

test_ratio=0.2
test_size=int(len(X) * test_ratio)

求出切分索引

test_indexes=shuffle_indexes[:test_size]
train_indexes=shuffle_indexes[test_size:]

获得数据

X_train=X[train_indexes]
y_train=y[train_indexes]

X_test=X[test_indexes]
y_test=y[test_indexes]

sklearn中的划分

有了之前的知识就能更好的理解sklearn中的切分函数的每个参数的意思。

导入包

from sklearn.model_selection import train_test_split

调用

X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.2, 
random_state=0)

参数意义：
X，数据集的输入
y, 数据集的输出
test_size 默认参数，这个测试集所占百分比。
radom_state 默认参数，随机数种子，一般调试的时候希望每次切分要是一样的，就给这个种子一个固定的值就好了。