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实现从百度API获取天气的方法
Mar 11 Python
python实现颜色rgb和hex相互转换的函数
Mar 19 Python
利用matplotlib+numpy绘制多种绘图的方法实例
May 03 Python
python使用正则表达式的search()函数实现指定位置搜索功能
Nov 10 Python
python实现log日志的示例代码
Apr 28 Python
python 列表递归求和、计数、求最大元素的实例
Nov 28 Python
Python3爬楼梯算法示例
Mar 04 Python
使用Python实现文字转语音并生成wav文件的例子
Aug 08 Python
python实现知乎高颜值图片爬取
Aug 12 Python
pytorch 固定部分参数训练的方法
Aug 17 Python
Python3将数据保存为txt文件的方法
Sep 12 Python
Python 如何实现访问者模式
Jul 28 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
浅析Yii中使用RBAC的完全指南(用户角色权限控制)
2013/06/20 PHP
php中url函数介绍及使用示例
2014/02/13 PHP
浅谈laravel-admin form中的数据,在提交后,保存前,获取并进行编辑
2019/10/21 PHP
javascript模拟的Ping效果代码 (Web Ping)
2011/03/13 Javascript
jQuery 删除/替换DOM元素的几种方式
2014/05/20 Javascript
Javascript定义类(class)的三种方法详解
2015/03/13 Javascript
浅谈Javascript实现继承的方法
2015/07/06 Javascript
Bootstrap的图片轮播示例代码
2015/08/31 Javascript
javascript实现拖动元素交换位置
2015/11/29 Javascript
JavaScript学习总结之JS、AJAX应用
2016/01/29 Javascript
JavaScript中localStorage对象存储方式实例分析
2017/01/12 Javascript
React简单介绍
2017/05/24 Javascript
JS中双击和单击事件冲突的解决方法
2018/04/09 Javascript
JavaScript设计模式之模板方法模式原理与用法示例
2018/08/07 Javascript
Nodejs监控事件循环异常示例详解
2019/09/22 NodeJs
Vue点击切换Class变化,实现Active当前样式操作
2020/07/17 Javascript
利用Fn.py库在Python中进行函数式编程
2015/04/22 Python
详解python的几种标准输出重定向方式
2016/08/15 Python
详解Python修复遥感影像条带的两种方式
2020/02/23 Python
python使用pymongo与MongoDB基本交互操作示例
2020/04/09 Python
pycharm无法安装第三方库的问题及解决方法以scrapy为例(图解)
2020/05/09 Python
使用OpenCV实现道路车辆计数的使用方法
2020/07/15 Python
canvas绘制视频封面的方法
2018/02/05 HTML / CSS
Farnell德国:电子元器件供应商
2018/07/10 全球购物
土木工程专业大学毕业生求职信
2013/10/13 职场文书
语文教育专业推荐信范文
2013/11/25 职场文书
本科毕业生专业自荐书范文
2014/02/05 职场文书
新学期决心书
2014/03/11 职场文书
《每逢佳节倍思亲》教后反思
2014/04/19 职场文书
最新优秀教师个人先进事迹材料
2014/05/06 职场文书
2014年学习部工作总结
2014/11/12 职场文书
2015年档案管理员工作总结
2015/05/13 职场文书
合同审查法律意见书
2015/06/04 职场文书
大学升旗仪式主持词
2015/07/04 职场文书
《7的乘法口诀》教学反思
2016/02/18 职场文书
MySQL深分页问题解决思路
2022/12/24 MySQL