码迷,mamicode.com
首页 > 其他好文 > 详细

数据集切分

时间:2020-02-20 23:56:20      阅读:117      评论:0      收藏:0      [点我收藏+]

标签:imp   random   code   value   val   dom   style   bsp   cti   

from sklearn.model_selection import train_test_split

Xtrain,Xtest,Ytrain,Ytest = train_test_split(X,Y,test_size=0.3,random_state=420)

#切分前数据标签的分布情况
train_data.SeriousDlqin2yrs.value_counts()

#切分后,训练集数据标签的分布情况
pd.Series(Ytrain).value_counts()

#如果切分后的数据集分布不均衡,则需要对Xtrain,Ytrain和Xtest,Ytest进行平衡处理(参见非平衡样本的处理方式SMOTE)

 

数据集切分

标签:imp   random   code   value   val   dom   style   bsp   cti   

原文地址:https://www.cnblogs.com/jing-yan/p/12339642.html

(0)
(0)
   
举报
评论 一句话评论(0
登录后才能评论!
© 2014 mamicode.com 版权所有  联系我们:gaon5@hotmail.com
迷上了代码!