train_test_split数据切分

Posted 51python

tags:

篇首语:本文由小常识网(cha138.com)小编为大家整理,主要介绍了train_test_split数据切分相关的知识,希望对你有一定的参考价值。

train_test_split 数据切分

格式:

X_train,X_test, y_train, y_test =cross_validation.train_test_split(train_data,train_target,test_size=0.3, random_state=0)

 

参数解释:
train_data:去除label的数据集!!!!
train_target:label集合!!!!
test_size:样本占比,如果是整数的话就是样本的数量
random_state:是随机数的种子。
随机数种子:其实就是该组随机数的编号,在需要重复试验的时候,保证得到一组一样的随机数。比如你每次都填1,其他参数一样的情况下你得到的随机数组是一样的。但填0或不填,每次都会不一样。
随机数的产生取决于种子,随机数和种子之间的关系遵从以下两个规则:种子不同,产生不同的随机数;种子相同,即使实例不同也产生相同的随机数。

 

样例:

原始数据data,最后一列为标签label(也可以放在第一行)

data=[[1 0 0 ... 1 0 6]
          [0 1 0 ... 0 0 9]

                    ...
          [0 0 0 ... 1 0 6]
          [0 1 0 ... 0 1 9]]

 

train_data=[[1 0 0 ... 1 0 ]
                  [0 1 0 ... 0 0 ]

          ...
                  [0 0 0 ... 1 0 ]
                  [0 1 0 ... 0 1 ]]

train_target=[6 9  ... 6 9]

 

利用切分函数train_test_split得到,训练集如下,测试集于此类似

X_train=[[1 0 0 ... 1 0 ]
               [0 1 0 ... 0 0 ]

                        ...
               [0 1 0 ... 0 1 ]]

y_train=[6 9 ... 9]

 

 

 

 

 

 

 

 

 

 


参考:https://blog.csdn.net/mrxjh/article/details/78481578 
















以上是关于train_test_split数据切分的主要内容,如果未能解决你的问题,请参考以下文章

模型评估:数据切分

分割训练集测试集验证集的两种方式

分割训练集测试集验证集的两种方式

最简单的机器学习流程 基于sklearn

关于train_test_split函数是不是打乱数据的验证

来自 train_test_split 和 seaborn 的可视化数据训练和数据测试