scikit-learn中的分层训练测试拆分

scikit-learn中的分层训练/测试拆分

[更新为0.17]

请参阅以下文档

sklearn.model_selection.train_test_split

：

from sklearn.model_selection import train_test_splitX_train, X_test, y_train, y_test = train_test_split(X, y,        stratify=y,         test_size=0.25)

[/更新为0.17]

有一个拉要求在这里。但是，您可以根据需要简单地进行

train, test =next(iter(StratifiedKFold(...)))

训练和测试索引的使用。