Python 实现训练集、测试集随机划分
作者：no_error_no_warn 发布时间：2023-02-12 01:41:24
标签：Python,训练集,测试集,划分
随机从列表中取出元素：
import random
dataSet = [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10]]
trainDataSet = random.sample(dataSet, 3)
以下函数，使用于我最近的一个机器学习的项目，将数据集数据按照比例随机划分成训练集数据和测试集数据：
import csv
import random
def getDataSet(proportion):
"""
:exception
获取训练集和测试集(将数据按比例随机划分)
:parameter
proportion - 测试集/数据集
:return
trainDataSet - 训练集
testDataSet - 测试集
author
肖政宇
modify
2019年5月10日
"""
dataSet = open('数据集.csv')
dataSetReader = csv.reader(dataSet)
"""
:exception
将数据保存到数组
"""
dataSet = []
next(dataSetReader, 'none') # 跳过表头
data = next(dataSetReader, 'none')
while (data != 'none'):
dataSet.append(data)
data = next(dataSetReader, 'none')
"""
:exception
按照比例随机划分出训练集和测试集
"""
dataNumber = dataSet.__len__() # 数据集数据条数
testNumber = int(dataNumber * proportion) # 测试集数据条数
testDataSet = [] # 测试数据集
trainDataSet = [] # 训练数据集
testDataSet = random.sample(dataSet, testNumber) # 测试集
for testData in testDataSet: # 将已经选定的测试集数据从数据集中删除
dataSet.remove(testData)
trainDataSet = dataSet # 训练集
return trainDataSet, testDataSet
来源：https://blog.csdn.net/qq_39514033/article/details/90052508
0
投稿猜你喜欢
自从接触python以后就想着爬pixiv，之前因为梯子有点问题就一直搁置，最近换了个梯子就迫不及待试了下。爬虫无非request获取htm- 这篇文章用来整理一下入门深度学习过程中接触到的四种激活函数，下面会从公式、代码以及图像三个方面介绍这几种激活函数，首先来明确一下是哪四种：S
- (一) 常用的CSS命名规则：头：header内容：content/container尾：footer导航：nav侧栏：sidebar栏目：
- 识别发票录入发票是一件繁琐的工作，如果可以自动识别并且录入系统，那可真是太好了。今天我们就来学习一下，如何自动识别增值税发票并且录入系统~识
- 本文实例讲述了JavaScript判断前缀、后缀是否是空格的方法。分享给大家供大家参考。具体如下：// Js 判断后缀 String.pro
- 画矩形函数调用:cv2.rectangle(img,pt1,pt2,color,thickness,line_type,shift)img:
- 众所周知IE6不支持position:fixed，这个bug与IE6的双倍margin和不支持PNG透明等bug一样臭名昭著。前些天我做自己
- 一： 基本使用：1：环境的安装：pip install flask-sqlalchemypip install pymysql2：组件初始化
- 今天刚好需要配置mysql 5.5.45，因为数据库量挺大的，所以必须优化，要不mysql真的不快。(1)、max_connections：
- PC登录新浪微博时，在客户端用js预先对用户名、密码都进行了加密，而且在POST之前会GET一组参数，这也将作为POST_DATA的一部分。
- 治標不治本的就是將php.ini內的reporting部份修改，讓notice不顯示 error_reporting = E_ALL; di
- 表结构如下面代码创建 CREATE TABLE test_tb ( TestId int not null identity(1,1) pr
- 本文实例讲述了Python实现扣除个人税后的工资计算器。分享给大家供大家参考，具体如下：正好处于找工作期间避免不了会跟单位谈论薪资的情况，当
- 一、图片验证码django-simple-captcha配置1.在pycharm中，File====》Settings====》Projec
- 简单的仿图片验证码，适合新手简单的仿图片验证码演示，很容易被破解，实用性不大，但拿出来给新手学习一下还是不错的：JScript.Asp代码示
- 查询所有数据库的总大小方法如下：mysql> use information_schema;mysql> select conc
- 在MySQL里，主键索引和辅助索引分别是什么意思，有什么区别？上次的分享我们介绍了聚集索引和非聚集索引的区别，本次我们继续介绍主键索引和辅助
- 一. MovingAverage权值滑动平均更新1.1 示例代码：def create_target_q_network(self,stat
- iframe的背景颜色在IE下为window的当前窗口颜色（默认白色，可以在显示外观处对其进行设置），这导致在大幅背景上插入iframe时，
- 删除单条记录 代码如下:id = saferequest("id") sql="
