在机器学习中,我们通常将原始数据按照比例分割为“测试集”和“训练集”,通常使用sklearn.cross_validation里的train_test_split模块用来分割数据。

cross_validation已经弃用,现在改为从 sklearn.model_selection 中调用train_test_split 函数。

简单用法如下:

X_train,X_test, y_train, y_test =sklearn.model_selection.train_test_split(train_data,train_target,test_size=0.4, random_state=0,stratify=y_train)

# train_data:所要划分的样本特征集

# train_target:所要划分的样本结果

# test_size:样本占比,如果是整数的话就是样本的数量

# random_state:是随机数的种子。

# 随机数种子:其实就是该组随机数的编号,在需要重复试验的时候,保证得到一组一样的随机数。比如你每次都填1,其他参数一样的情况下你得到的随机数组是一样的。但填0或不填,每次都会不一样。

stratify是为了保持split前类的分布。比如有100个数据,80个属于A类,20个属于B类。如果train_test_split(... test_size=0.25, stratify = y_all), 那么split之后数据如下: 
training: 75个数据,其中60个属于A类,15个属于B类。 
testing: 25个数据,其中20个属于A类,5个属于B类。

用了stratify参数,training集和testing集的类的比例是 A:B= 4:1,等同于split前的比例(80:20)。通常在这种类分布不平衡的情况下会用到stratify。

将stratify=X就是按照X中的比例分配

将stratify=y就是按照y中的比例分配

train_test_split参数含义相关推荐

  1. sklearn的train_test_split()各函数参数含义解释(非常全)

    sklearn的train_test_split()各函数参数含义解释(非常全) sklearn的train_test_split()各函数参数含义解释(非常全) - The-Chosen-One - ...

  2. DeepCTR-Torch 如何使用【案例(Criteo、Movielens)演示、特征(SparseFeat、DenseFeat、VarLenSparseFeat)参数含义】

    文章目录 安装 DeepCTR-Torch 的 4 个步骤 step1:模型导入 step2:简单的预处理 step3:生成特征列 step4:生成训练样本并训练模型 分类:Criteo 回归:Mov ...

  3. Java Calendar.add()方法的使用,参数含义。指定时间差。

    cal.add()方法中的参数含义: 第一个参数如果是1则代表的是对年份操作,2是对月份操作,3是对星期操作,5是对日期操作,11是对小时操作,12是对分钟操作,13是对秒操作,14是对毫秒操作. 第 ...

  4. python函数type的用意_Python中type的构造函数参数含义说明

    测试代码如下: 复制代码 代码如下: class ModelMetaClass(type): def __new__(cls,name,base,attrs): logging.info(" ...

  5. paip.提升效率--调试--日志系统日志参数含义---python

    paip.提升效率--调试--日志系统日志参数含义---python #同时向控制台和文件输出日志 #日志参数含义 import logging log_format = '%(filename)s ...

  6. OpenGL透视投影函数gluPerspective的参数含义之解析(图示)

    首先解释一下透视投影: **透视投影属于中心投影.**透视投影图简称为透视图或透视,它是从某个投射中心将物体投射到单一投影面上所得到的图形.透视图与人们观看物体时所产生的视觉效果非常接近,所以它能更加 ...

  7. 陶瓷气体放电管参数含义详解

    ​很多客户反应,不太明白陶瓷气体放电管产品手册中的参数含义.不可否认,电路保护器件产品规格书手册用的语言大部分都是英文,没有一定的英文基础,还真消化不了.有时候,就算能看得懂,但是面对枯燥无味的参数, ...

  8. GoldenGate 基本参数含义

    GoldenGate 基本参数含义 1. edit params命令 2. DBLOGIN USERID <user>, PASSWORD <password> 3. ADD ...

  9. cursor is oracle 日期_Oracle时间计算------日期格式参数含义说明

    Oracle时间计算------日期格式参数含义说明 D 一周中的星期几 DAY 天的名字,使用空格填充到9个字符 DD 月中的第几天 DDD 年中的第几天 DY 天的简写名 IW ISO标准的年中的 ...

最新文章

  1. ndarray.shape[]返回值的意义
  2. 有人喜欢爱情测试小游戏么
  3. 数据分析Python:sklearn数据预处理中fit(),transform()与fit_transform()的区别
  4. django 日志配置
  5. linux安装Git依赖的包出错,技术|Linux有问必答:如何在Linux上安装Git
  6. 2021年快手大健康行业数据价值报告
  7. 816 - Abbott's Revenge
  8. Mouse Event (Java AWT)
  9. Android中通过ActionBar为标题栏添加搜索以及分享视窗
  10. 博文视点“阿里云全系列技术图书”隆重亮相2021云栖大会
  11. 华为云计算hcip证书有效期_华为云计算容器HCIP V1.0认证
  12. svn切换项目地址(外网 > 内网)
  13. Proxomitron之天涯只看规则
  14. 诊断故障码(Diagnostic Trouble Code-DTC)
  15. 支付宝异步回调步骤+内网穿透++雷神商城项目
  16. 达人评测 i5-1155g7和i5-1235u选哪个 i51155g7和1235u差距
  17. 数据预处理中的归一化与反归一化
  18. linux环境搭建篇-redis设置密码
  19. AirVO: An Illumination-Robust Point-Line Visual Odometry阅读
  20. coda html5,Mac Coda 前端开发插件 F2E Tools - YangJunwei

热门文章

  1. 钱都花哪去了?预算费用控制管理系统帮企业精准管理“金脉”
  2. vue 项目启动报错“Cannot GET /”
  3. OCR光学字符识别(一)
  4. SQL语句-各种查询
  5. IPv6的HappyEyeballs算法
  6. 楼宇能效控制器真的有用吗
  7. 智能可视化门铃方案调研报告
  8. 必备知识---TCP三次握手和四次挥手以及SSL四次握手
  9. oracle找回删除的表
  10. 字节数组与String类型的转换