SQL中如何利用ROW_NUMBER实现测试集与训练集拆分

夏静大大_7547

夏静大大_7547

2026-10-24

297人浏览

原创

row_number()能用于训练/测试集拆分,因其可为每行稳定、可复现地生成唯一稠密序号;需显式order by(如id或created_at)确保一致性,再结合mod取模(如%10)或ntile分桶实现比例拆分,并须用cte或子查询一次性标记以保证互斥。

sql中如何利用row_number实现测试集与训练集拆分

ROW_NUMBER() 为什么能用来做训练/测试集拆分

因为 ROW_NUMBER() 能给每一行稳定、可复现地打上唯一序号,配合取模或范围筛选,就能把数据按比例切开。它不依赖数据本身分布,也不需要随机种子(但要注意:默认无 ORDER BY 时结果不可靠)。

  • 必须显式写 ORDER BY,否则不同执行可能生成不同序号,导致训练集“每次都不一样”
  • 用 ORDER BY RAND() 看似随机,但在 MySQL 8.0+ 和 PostgreSQL 中性能差、不可复现;SQL Server 的 NEWID() 同理
  • 更稳妥的做法是选一个业务上稳定且有区分度的字段排序,比如 id、created_at,再结合取模或窗口截断

用 MOD 取模实现近似 8:2 拆分(推荐初用)

这是最轻量、兼容性最好的方式,适用于大多数 OLAP 场景。核心思路:按某字段排序后编号,再对编号取模,把余数为 0~7 的归训练集,8~9 归测试集。

SELECT *,
       ROW_NUMBER() OVER (ORDER BY id) % 10 AS split_mod
FROM your_table;
  • 后续加 WHERE split_mod 得训练集,<code>WHERE split_mod IN (8, 9) 得测试集
  • 如果表里 id 有空缺或不连续,不影响比例稳定性——因为 ROW_NUMBER() 生成的是稠密序号
  • 注意:MySQL 5.7 不支持直接在 WHERE 中引用窗口函数别名,得套一层子查询或 CTE

用 NTILE(10) 做等频分桶(适合倾斜数据)

当原始数据按时间或 ID 分布不均(比如最近一周数据暴增),单纯用 MOD 可能导致测试集集中在某个时间段。NTILE(10) 会尽量让每组行数相等,再挑其中两组作测试集更鲁棒。

SELECT *,
       NTILE(10) OVER (ORDER BY created_at) AS bucket
FROM your_table;
  • NTILE 不保证绝对均匀(总行数不能被 10 整除时,前面几组多一行),但比 MOD 对偏斜更友好
  • 选 bucket IN (1, 2) 还是 bucket IN (9, 10) 取决于你是否想让测试集覆盖首尾分布——通常建议避开两端,选中间两组(如 4 和 5)以减少时序泄露风险
  • PostgreSQL 和 SQL Server 支持,但 SQLite 不支持 NTILE

训练集和测试集必须严格互斥且无 overlap

最容易被忽略的一点:拆分逻辑必须作用于同一份基础数据快照。如果两次查询分别执行 ROW_NUMBER(),又没加 WITH TIES 或事务隔离,可能因并发写入导致某条记录同时出现在两个集合里。

  • 务必用 CTE 或子查询一次性算出所有标记列,再分别 SELECT
  • 例如:先 WITH labeled AS (SELECT *, ROW_NUMBER() OVER (...) % 10 AS mod_val FROM t),再 SELECT * FROM labeled WHERE mod_val 和 <code>SELECT * FROM labeled WHERE mod_val >= 8
  • 如果底层表正在被 ETL 写入,记得加 FOR SHARE(PostgreSQL)或指定事务隔离级别,否则可能读到部分更新状态
PHP速学视频免费教程(入门到精通)
PHP速学视频免费教程(入门到精通)

PHP怎么学习?PHP怎么入门?PHP在哪学?PHP怎么学才快?不用担心,这里为大家提供了PHP速学教程(入门到精通),有需要的小伙伴保存下载就能学习啦!

下载

相关标签:

本站声明:本文内容由网友自发贡献,版权归原作者所有,本站不承担相应法律责任。如您发现有涉嫌抄袭侵权的内容,请联系admin@php.cn

相关专题

更多
数据分析工具有哪些
数据分析工具有哪些

数据分析工具有Excel、SQL、Python、R、Tableau、Power BI、SAS、SPSS和MATLAB等。详细介绍:1、Excel,具有强大的计算和数据处理功能;2、SQL,可以进行数据查询、过滤、排序、聚合等操作;3、Python,拥有丰富的数据分析库;4、R,拥有丰富的统计分析库和图形库;5、Tableau,提供了直观易用的用户界面等等。

2023.10.12

4463

8

SQL中distinct的用法
SQL中distinct的用法

SQL中distinct的语法是“SELECT DISTINCT column1, column2,...,FROM table_name;”。本专题为大家提供相关的文章、下载、课程内容,供大家免费下载体验。

2023.10.27

931

4

SQL中months_between使用方法
SQL中months_between使用方法

在SQL中,MONTHS_BETWEEN 是一个常见的函数,用于计算两个日期之间的月份差。想了解更多SQL的相关内容,可以阅读本专题下面的文章。

2024.02.23

1169

5

SQL出现5120错误解决方法
SQL出现5120错误解决方法

SQL Server错误5120是由于没有足够的权限来访问或操作指定的数据库或文件引起的。想了解更多sql错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

6421

10

sql procedure语法错误解决方法
sql procedure语法错误解决方法

sql procedure语法错误解决办法:1、仔细检查错误消息;2、检查语法规则;3、检查括号和引号;4、检查变量和参数;5、检查关键字和函数;6、逐步调试;7、参考文档和示例。想了解更多语法错误的相关内容,可以阅读本专题下面的文章。

2024.03.06

3223

4

oracle数据库运行sql方法
oracle数据库运行sql方法

运行sql步骤包括:打开sql plus工具并连接到数据库。在提示符下输入sql语句。按enter键运行该语句。查看结果,错误消息或退出sql plus。想了解更多oracle数据库的相关内容,可以阅读本专题下面的文章。

2024.04.07

6380

11

sql中where的含义
sql中where的含义

sql中where子句用于从表中过滤数据,它基于指定条件选择特定的行。想了解更多where的相关内容,可以阅读本专题下面的文章。

2024.04.29

8721

6

sql中删除表的语句是什么
sql中删除表的语句是什么

sql中用于删除表的语句是drop table。语法为drop table table_name;该语句将永久删除指定表的表和数据。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

1190

5

sql中删除一列的命令是什么
sql中删除一列的命令是什么

在sql中,使用alter table语句可以删除一列,语法为:alter table table_name drop column column_name。想了解更多sql的相关内容,可以阅读本专题下面的文章。

2024.04.29

1032

5

热门下载

更多
网站特效
/
网站源码
/
网站素材
/
前端模板

精品课程

更多
热门推荐
/
最新课程
phpStudy极速入门视频教程
phpStudy极速入门视频教程

共6课时 | 54.7万人学习

独孤九贱(4)_PHP视频教程
独孤九贱(4)_PHP视频教程

共89课时 | 133.9万人学习