AI 漫天,我们坚持手做论坛。

新加坡 10-30 20:45 · 吉隆坡 20:45 · 雅加达 19:45 · 曼谷 19:45 · 马尼拉 20:45 · 胡志明 19:45 · 北京 20:45 · 东京 21:45 · 伦敦 12:45 · 纽约 08:45

GroupShuffleSplit的test_size按组计算,测试样本比例未必等于设定比例

#机器学习#数据划分#教学

1
新加坡 · 中文 · r/人工智能 · Amy · 声望 463 · 10-30 19:21 新 · 帖子

Amy 原创整理|资料核验:2026-10-30

scikit-learn的GroupShuffleSplit按给定分组随机划分训练与测试数据。test_size和train_size指的是组,而不是样本行数;test_size为小数时,测试组数按比例向上取整。

教学例子:10台设备分别产生不同数量的记录,test_size=0.2表示每次选择2台设备对应的组作测试,而不保证测试记录恰好占全部记录的20%。如果设备记录量差异大,两种比例可能差很多。

Amy建议每次划分后同时记录组数和样本数。文档还提醒,多次随机划分得到的测试集可能互相重叠;不要把n_splits次结果当作互不重叠的完整分区。报告分数时,把划分方式也写清楚。

官方来源:
https://scikit-learn.org/stable/modules/generated/sklearn.model_selection.GroupShuffleSplit.html

回复

还没有回复。

广告由 Google AdSense 投放,标为广告,与帖子分开。请勿点击自己的广告,也不要请别人点。行情讨论不是荐股。本地服务帖写地点和价格,线下交易注意安全。

关于 · 隐私 · 联系 · 版规