Python 实现训练集、测试集随机划分


Posted in Python onJanuary 08, 2020

随机从列表中取出元素:

import random
dataSet = [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10]]
trainDataSet = random.sample(dataSet, 3)

Python 实现训练集、测试集随机划分

以下函数,使用于我最近的一个机器学习的项目,将数据集数据按照比例随机划分成训练集数据和测试集数据:

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

以上这篇Python 实现训练集、测试集随机划分就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持三水点靠木。

Python 相关文章推荐
python 调用HBase的简单实例
Dec 18 Python
Python 实现购物商城,含有用户入口和商家入口的示例
Sep 15 Python
python使用logging模块发送邮件代码示例
Jan 18 Python
Python使用win32 COM实现Excel的写入与保存功能示例
May 03 Python
Python中实例化class的执行顺序示例详解
Oct 14 Python
Python3实现对列表按元组指定列进行排序的方法分析
Dec 22 Python
python ddt数据驱动最简实例代码
Feb 22 Python
Python魔法方法功能与用法简介
Apr 04 Python
python基于plotly实现画饼状图代码实例
Dec 16 Python
jupyter notebook 恢复误删单元格或者历史代码的实现
Apr 17 Python
浅谈python3打包与拆包在函数的应用详解
May 02 Python
python 实现aes256加密
Nov 27 Python
Pyecharts绘制全球流向图的示例代码
Jan 08 #Python
PyTorch 解决Dataset和Dataloader遇到的问题
Jan 08 #Python
使用PyTorch将文件夹下的图片分为训练集和验证集实例
Jan 08 #Python
使用 PyTorch 实现 MLP 并在 MNIST 数据集上验证方式
Jan 08 #Python
Pycharm 2020最新永久激活码(附最新激活码和插件)
Sep 17 #Python
将matplotlib绘图嵌入pyqt的方法示例
Jan 08 #Python
pyinstaller还原python代码过程图解
Jan 08 #Python
You might like
php mssql 日期出现中文字符的解决方法
2009/03/10 PHP
PHP简单遍历对象示例
2016/09/28 PHP
PHP实现数组向任意位置插入,删除,替换数据操作示例
2019/04/05 PHP
JQuery的$和其它JS发生冲突的快速解决方法
2014/01/24 Javascript
js取整数、取余数的方法
2014/05/11 Javascript
js中常用的Tab切换效果(推荐)
2016/08/30 Javascript
浅析JavaScript中作用域和作用域链
2016/12/06 Javascript
VUE中使用Vue-resource完成交互
2017/07/21 Javascript
EasyUI在Panel上动态添加LinkButton按钮
2017/08/11 Javascript
ES6中的Promise代码详解
2017/10/09 Javascript
JS双向链表实现与使用方法示例(增加一个previous属性实现)
2019/01/31 Javascript
将RGB值转换为灰度值的简单算法
2019/10/09 Javascript
用Node写一条配置环境的指令
2019/11/14 Javascript
微信小程序跨页面传递data数据方法解析
2019/12/13 Javascript
vue组件实现移动端九宫格转盘抽奖
2020/10/16 Javascript
Python实现按特定格式对文件进行读写的方法示例
2017/11/30 Python
python3利用smtplib通过qq邮箱发送邮件方法示例
2017/12/03 Python
Python实现定时备份mysql数据库并把备份数据库邮件发送
2018/03/08 Python
Tensorflow卷积神经网络实例
2018/05/24 Python
python使用pygame实现笑脸乒乓球弹珠球游戏
2019/11/25 Python
python基于TCP实现的文件下载器功能案例
2019/12/10 Python
简单了解如何封装自己的Python包
2020/07/08 Python
Python模块zipfile原理及使用方法详解
2020/08/04 Python
python3获取控制台输入的数据的具体实例
2020/08/16 Python
python 中 .py文件 转 .pyd文件的操作
2021/03/04 Python
Kate Spade美国官网:纽约新兴时尚品牌,以包包闻名于世
2017/11/09 全球购物
美国领先的家庭健康检测试剂盒提供商:LetsGetChecked
2019/03/18 全球购物
小学教学随笔感言
2014/02/26 职场文书
项目合作意向书范本
2014/04/01 职场文书
继承公证书样本
2014/04/04 职场文书
工人先进事迹材料
2014/12/26 职场文书
通讯稿格式及范文
2015/07/22 职场文书
2016会计专业自荐信范文
2016/01/28 职场文书
详细总结Python常见的安全问题
2021/05/21 Python
Python访问Redis的详细操作
2021/06/26 Python
MySQL表锁、行锁、排它锁及共享锁的使用详解
2022/04/02 MySQL