码迷,mamicode.com
首页 > 其他好文 > 详细

train_test_split数据切分

时间:2018-12-25 23:33:55      阅读:206      评论:0      收藏:0      [点我收藏+]

标签:stat   https   dom   数据集   利用   state   blog   解释   and   

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数据切分

标签:stat   https   dom   数据集   利用   state   blog   解释   and   

原文地址:https://www.cnblogs.com/51python/p/10176890.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!