如何将数据分成3组(训练、验证和测试)？

Question

如何将数据分成3组(训练、验证和测试)？我有一只熊猫数据，我想把它分成3组。我知道用列车试验劈开从…sklearn.cross_validation，可以将数据分成两组(训练和测试)。但是，我无法找到将数据分成三组的解决方案。最好，我想要原始数据的索引。我知道解决办法是用train_test_split两次调整指数。但是，是否有更标准/内置的方式将数据分割成3组而不是2组？

Answer 1

但是，有一种方法可以将数据集划分为train, test, cv带着0.6, 0.2, 0.2将使用train_test_split方法两次。

from sklearn.model_selection import train_test_split

x, x_test, y, y_test = train_test_split(xtrain,labels,test_size=0.2,train_size=0.8)x_train, x_cv, y_train, y_cv = train_test_split(x,y,test_size = 0.25,train_size =0.75)