train_test_split 与 StratifiedShuffleSplit 中的分层参数

Roh*_*nto 5 scikit-learn train-test-split

在sklearn函数stratify中使用参数和函数有什么区别?他们不做同样的事情吗?train_test_splitStratifiedShuffleSplit

des*_*aut 4

这两个模块执行不同的操作。

train_test_split顾名思义,用于将数据拆分为单个训练和单个测试子集,并且该stratify参数允许以分层方式执行此操作。

StratifiedShuffleSplit另一方面,为交叉验证提供分割;来自文档

分层ShuffleSplit交叉验证器

提供训练/测试索引来分割训练/测试集中的数据

注意复数集合(强调我的)。

那么,当我们想要确保 CV 分割是分层的而不是替换 时,StratifiedShuffleSplit是否可以使用.KFoldtrain_test_split