[更新为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(...)))训练和测试索引的使用。
