train_test_split的用法(机器学习)

机器学习中对于数据的随机划分

train_test_split的引用: from sklearn.model_selection import train_test_split

例如:

import numpy as np

from sklearn.model_selection import train_test_split

X,y=np.arange(10).reshape((5,2)),range
X=np.array([[0,1],[2,3],[4,5],[6,7],[8,9]])
y=[0,1,2,3,4]
print(X)
print(y)

X_train,X_test,y_train,y_test=train_test_split(X,y,test_size=0.4,random_state=0)

print(X_train)

print(X_test)

print(y_train)

print(y_test)

#X_train为训练集,X_test为测试集,y_train为训练集,y_test为测试集

train_test_split  将数据集划分成X_train,X_test,y_train,y_test  四部分

train_test_split (train_data,train_target,test_size=0.4, random_state=0)

train_data:所要划分的样本整体

train_target所要划分的样本分类结果

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

random_state=0:随机数种子  这里的random_state就是为了保证程序每次运行都分割一样的训练集和测试集。   

思考题:random=0和random=1有什么区别?

  • 3
    点赞
  • 8
    收藏
    觉得还不错? 一键收藏
  • 1
    评论

“相关推荐”对你有帮助么?

  • 非常没帮助
  • 没帮助
  • 一般
  • 有帮助
  • 非常有帮助
提交
评论 1
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

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

抵扣说明:

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

余额充值