使用train_test_split 分割测试集以及训练集

这篇博客介绍了如何利用Python的sklearn库中的train_test_split函数将数据集划分为训练集和测试集,以进行机器学习模型的训练和验证。通过设定test_size参数为0.33,随机种子为42,确保了数据划分的稳定性和测试集占总数据的33%。

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

最新训练模型需要分别把数据分成训练集和测试集,用到了这个,记录下

from sklearn.model_selection import train_test_split    

X=df.iloc[:,0:4].values.tolist()
y=df.iloc[:,4]
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.33, random_state=42)

 

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值