sklearn数据集随机切分（train_test_split）

最新推荐文章于 2025-01-21 18:13:29 发布

竹聿Simon

最新推荐文章于 2025-01-21 18:13:29 发布

阅读量3.3w

点赞数 4

分类专栏： Python sklearn Python学习文章标签： python skleran 数据集切分测试集

本文链接：https://blog.csdn.net/churximi/article/details/61415254

版权

这篇博客介绍了如何使用sklearn库将数据集按一定比例随机划分为训练集和测试集，确保标签的一一对应。文中给出了更新后的代码示例。

摘要生成于 C知道，由 DeepSeek-R1 满血版支持，前往体验 >

sklearn学习

给定数据集X和类别标签y，将数据集按一定比例随机切分为训练集和测试集。

代码

#!/usr/bin/env python
# -*- coding: utf-8 -*-

"""
功能：数据集按比例切分为训练集和测试集
时间：2017年3月11日 12:48:57
"""

# from sklearn.cross_validation import train_test_split
from sklearn.model_selection import train_test_split  # 更新

# 生成200个句子，前100个和后100个类别分别对应1和2
X = [[u"这是", u"第1个", u"测试"]] * 100 + [[u"这是", u