python 划分数据集为训练集和测试集的方法

更新时间：2018年12月11日 15:29:50 作者：心雨心辰

今天小编就为大家分享一篇python 划分数据集为训练集和测试集的方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧

sklearn的cross_validation包中含有将数据集按照一定的比例，随机划分为训练集和测试集的函数train_test_split

from sklearn.cross_validation import train_test_split
#x为数据集的feature熟悉，y为label.
x_train, x_test, y_train, y_test = train_test_split(x, y, test_size = 0.3)

得到的x_train,y_train（x_test,y_test）的index对应的是x,y中被抽取到的序号。

若train_test_split传入的是带有label的数据，则如下代码：

from sklearn.cross_validation import train_test_split
#dat为数据集,含有feature和label.
train, test = train_test_split(dat, test_size = 0.3)

train,test含有feature和label的。

自己写了一个函数：

#X:含label的数据集：分割成训练集和测试集
#test_size:测试集占整个数据集的比例
def trainTestSplit(X,test_size=0.3):
 X_num=X.shape[0]
 train_index=range(X_num)
 test_index=[]
 test_num=int(X_num*test_size)
 for i in range(test_num):
  randomIndex=int(np.random.uniform(0,len(train_index)))
  test_index.append(train_index[randomIndex])
  del train_index[randomIndex]
 #train,test的index是抽取的数据集X的序号
 train=X.ix[train_index] 
 test=X.ix[test_index]
 return train,test

以上这篇python 划分数据集为训练集和测试集的方法就是小编分享给大家的全部内容了，希望能给大家一个参考，也希望大家多多支持脚本之家。

您可能感兴趣的文章:

Python+Turtle实现绘制可爱的小仓鼠
肉嘟嘟的小动物很是可爱，这篇文章主要为大家介绍一下如何运用Python中的turtle库控制函数绘制小仓鼠，文中的实现方法讲解详细，感兴趣的可以尝试一下
2022-10-10
如何给windows设置定时任务并运行python脚本
这篇文章主要介绍了如何给windows设置定时任务并运行python脚本，本文给大家介绍的非常详细，对大家的学习或工作具有一定的参考借鉴价值，需要的朋友可以参考下
2021-08-08
教你使用一行Python代码玩遍童年的小游戏
这篇文章主要介绍了一行Python代码玩遍童年的小游戏，帮助大家重回童年快乐时光，代码简单易懂，感兴趣的朋友一起学习下吧
2021-08-08
python cx_Oracle的基础使用方法(连接和增删改查)
这篇文章主要给大家介绍了关于python cx_Oracle的基础使用方法，其中包括连接、增删改查等基本操作，并给大家分享了python 连接Oracle 乱码问题的解决方法，需要的朋友可以参考借鉴，下面随着小编来一起学习学习吧。
2017-11-11
pytorch 搭建神经网路的实现
这篇文章主要介绍了pytorch 搭建神经网路，文中通过示例代码介绍的非常详细，具有一定的参考价值，感兴趣的小伙伴们可以参考一下
2021-08-08
pytorch无法使用GPU问题的解决方法
这篇文章主要介绍了如何解决pytorch 无法使用GPU 的问题,文中通过代码和图文给大家讲解的非常详细,对大家的学习或工作有一定的帮助,需要的朋友可以参考下
2024-02-02
在python中logger setlevel没有生效的解决
今天小编就为大家分享一篇在python中logger setlevel没有生效的解决，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧
2020-02-02
Python接口自动化之文件上传/下载接口详解
在接口测试中,经常会涉及到文件上传以及下载,下面这篇文章主要给大家介绍了关于Python接口自动化之文件上传/下载接口的相关资料,文中通过示例代码介绍的非常详细,需要的朋友可以参考下
2022-04-04
Flask框架通过Flask_login实现用户登录功能示例
这篇文章主要介绍了Flask框架通过Flask_login实现用户登录功能,结合实例形式较为详细的分析了flask框架使用Flask_login实现用户登陆功能的具体操作步骤、相关实现技巧与操作注意事项,需要的朋友可以参考下
2018-07-07
Python 语法错误:"SyntaxError: invalid charac
本文给大家分享Python 语法错误：“SyntaxError: invalid character in identifier“，原因及解决方法，文末给大家补充介绍了Python出现SyntaxError: invalid syntax的原因总结，感兴趣的朋友跟随小编一起学习吧
2023-02-02