python 划分数据集为训练集和测试集的方法
作者：心雨心辰 发布时间：2023-01-10 12:18:36
标签：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
来源：https://blog.csdn.net/xidianliutingting/article/details/53463033
0
投稿猜你喜欢
在Goland中，如果 import 了包，但在代码中没有使用，会自动帮你移除这个包的 引用有可能是习惯问题，每次写代码都习惯 先impor- 本文实例讲述了C#简单连接sql数据库的方法。分享给大家供大家参考，具体如下：using System;using System.Colle
- 简单的测试用例#!/usr/bin/python# -*- coding: UTF-8 -*-import timefrom pymongo
- logging模块是Python内置的标准模块，主要用于输出运行日志，可以设置输出日志的等级、日志保存路径、日志文件回滚等。其主
- Python脚本编辑使用Python对fasta格式的序列进行基本信息统计预期设计输出文件中包括fasta文件名，序列长度，GC含量以及AT
- 背景：在pycharm中运行正常，但是使用命令方式就提示没有模块解决方案在Python安装目录下的\Lib\site-packages文件夹
- 想要在准备查询混合数据库系统时将SQL Server转换为 XQuery 和进行反向转换吗?了解如何开发 Web 服务来从SQL Serve
- whoosh的官方介绍：http://whoosh.readthedocs.io/en/latest/quickstart.html因为做的
- 下面就是简单的例子，这里提供2中方法：test.htm &
- 本文实例讲述了Python列表生成式与生成器操作。分享给大家供大家参考，具体如下：列表生成式：能够用来创建list的生成式比如想要生成类似[
- 开源方：微软安装：pip install playwright；python -m playwright install特点：自动化脚本录制
- 当我想要完美的使用:nth-child或者:nth-of-type的时候有点儿头晕。你越理解它们，就能写出越好的CSS规则！在这些简单的”秘
- 视图函数中加上认证功能，流程见下图import hashlibimport timedef get_random(name):
- 功能1： 爬取西拉ip代理官网上的代理ip环境：python3.8+pycharm库：requests，lxml浏览器：谷歌IP地址：htt
- 我们来学习一下 Python 中的加密模块，加密模块在工作中被广泛应用。比如数据的传入 不希望被捕获，通过把数据加密。这样即使被捕获也无法获
- 本文实例讲述了Python使用pymongo模块操作MongoDB的方法。分享给大家供大家参考，具体如下：通过pymongo实现python
- 本文实例为大家分享了python实现通讯录系统的具体代码，供大家参考，具体内容如下一、首先，给大家看一下这个系统的主菜单效果如下：二、步骤分
- <html><head><meta http-equiv="Content-T
- <?php $curDomain = $_SERVER['HTTP_HOST']; $strHTML = file_g
- 本文实例讲述了Python计时相关操作。分享给大家供大家参考，具体如下：内容目录：1. 时间戳2. 当前时间3. 时间差4. python中
