数据集切分
Posted jing-yan
tags:
篇首语：本文由小常识网(cha138.com)小编为大家整理，主要介绍了数据集切分相关的知识，希望对你有一定的参考价值。
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)
以上是关于数据集切分的主要内容，如果未能解决你的问题，请参考以下文章