返回分层的训练集和测试集,sklearn-learn 中的StratifiedShuffleSplit
Deprecated since version 0.18: This
module will be removed in 0.20. Usesklearn.model_selection.StratifiedShuffleSplit
instead.
0.18之前版本使用上面那种方式,新的版本使用下面那种方式。
旧类的使用:
sklearn.cross_validation.
StratifiedShuffleSplit
(y, n_iter=10, test_size=0.1, train_size=None, random_state=None)
新类的使用:
sklearn.model_selection.
StratifiedShuffleSplit
(n_splits=10, test_size=’default’, train_size=None, random_state=None)