train_test_split()函数划分训练、测试数据

说在前面

train_test_split()函数划分训练、测试集的用法

具体使用

train_test_split()函数是用来随机划分样本数据为训练集和测试集的,也可以用来人为的切片划分

可以客观随机的划分数据,减少认为因素

使用模板:
train_X,test_X,train_Y,test_Y=train_test_split(train_data,train_target,test_size=0.2,random_state=5)

参数解释:

  • train_data:待划分样本数据
  • train_target:待划分样本数据的结果(标签)
  • test_size:测试数据占样本数据的比例,若整数则是样本数据
  • random_state:设置随机数种子,保证每次都是同一个随机数。若为0或者不填,则每次得到的数据都不一样
posted @ 2021-01-13 17:58  酸奶面包  阅读(875)  评论(0编辑  收藏  举报