热线电话:13121318867

登录
2020-07-14 阅读量: 911
python:在sklearn中如何切分数据集

问题:为了得到更好的模型效果,我们往往需要将数据集分成训练集和测试集,那么具体怎么切分呢?

解答:

代码:##数据集

image.png

##切分数据集代码

from sklearn.model_selection import train_test_split


X_train, X_test, y_train, y_test = train_test_split(X, # array或者是DF

y, # 标签

test_size=0.3, # 切分出来的测试集的占比

random_state=42# 随机数种子

)

##一般测试集和训练集的比例是三七分


13.7937
2
关注作者
收藏
评论(0)

发表评论

暂无数据
推荐帖子