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中self的用法
Jun 04 Python
解决python3 pika之连接断开的问题
Dec 18 Python
Django2.1集成xadmin管理后台所遇到的错误集锦(填坑)
Dec 20 Python
在Python中将函数作为另一个函数的参数传入并调用的方法
Jan 22 Python
Django使用AJAX调用自己写的API接口的方法
Mar 06 Python
深入了解Python iter() 方法的用法
Jul 11 Python
在pycharm下设置自己的个性模版方法
Jul 15 Python
Django REST framework内置路由用法
Jul 26 Python
Pyecharts地图显示不完成问题解决方案
May 11 Python
Python-jenkins模块获取jobs的执行状态操作
May 12 Python
tensorflow中的梯度求解及梯度裁剪操作
May 26 Python
尝试使用Python爬取城市租房信息
Apr 12 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木马webshell扫描器代码
2012/01/25 PHP
使用PHP强制下载PDF文件示例
2014/01/17 PHP
jquery tools之tooltip
2009/07/25 Javascript
JavaScript中使用replace结合正则实现replaceAll的效果
2010/06/04 Javascript
用JS判别浏览器种类以及IE版本的几种方法小结
2011/08/02 Javascript
基于jquery库的tab新形式使用
2012/11/16 Javascript
不同的jQuery API来处理不同的浏览器事件
2012/12/09 Javascript
可选择和输入的下拉列表框示例
2013/11/05 Javascript
使用js判断TextBox控件值改变然后出发事件
2014/03/07 Javascript
js控制div弹出层实现方法
2015/05/11 Javascript
浅谈jquery页面初始化的4种方式
2016/11/27 Javascript
微信小程序 wx.request方法的异步封装实例详解
2017/05/18 Javascript
Angular 容器部署的方法
2018/04/17 Javascript
浅谈ECMAScript 中的Array类型
2019/06/10 Javascript
Vue.js项目实战之多语种网站的功能实现(租车)
2019/08/07 Javascript
JS求解两数之和算法详解
2020/04/28 Javascript
vue-router 2.0 跳转之router.push()用法说明
2020/08/12 Javascript
python链接Oracle数据库的方法
2015/06/28 Python
python使用ddt过程中遇到的问题及解决方案【推荐】
2018/10/29 Python
如何使用django的MTV开发模式返回一个网页
2019/07/22 Python
tensorflow获取预训练模型某层参数并赋值到当前网络指定层方式
2020/01/24 Python
Python3将ipa包中的文件按大小排序
2020/04/17 Python
Python如何把十进制数转换成ip地址
2020/05/25 Python
基于keras中的回调函数用法说明
2020/06/17 Python
html5录音功能实战示例
2019/03/25 HTML / CSS
英国时尚饰品和发饰购物网站:Claire’s
2017/07/04 全球购物
阿玛瑞酒店中文官方网站:Amari.com
2018/02/13 全球购物
泰国健康和美容服务预订网站:GoWabi
2019/06/03 全球购物
大学生学习生活的自我评价
2013/11/01 职场文书
畜牧兽医本科生的自我评价
2014/03/03 职场文书
教师中国梦演讲稿
2014/04/23 职场文书
毕业生求职信范文
2014/06/29 职场文书
离婚被告答辩状
2015/05/22 职场文书
导游词之黄果树瀑布
2019/09/20 职场文书
react 项目中引入图片的几种方式
2021/06/02 Javascript
MYSQL如何查看进程和kill进程
2022/03/13 MySQL