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…
train_test_split()详解
最新推荐文章于 2024-07-19 03:14:54 发布