cjdwd的博客

原创 train_test_split()详解

train_X1,test_X1,train_X2,test_X2,… = train_test_split(X1,X2,…test_size=0.3,random_state=5,stratify=X1)可以导入任意多个数据集X1,X2…test_size是test集占总集的比例random_state是随机数种子stratify是分层对象，其余的对应着这个分层返回分好的X1,X2…...