首頁猿問如何將數(shù)據(jù)分成3組(訓(xùn)練、驗(yàn)證和測(cè)試)？

如何將數(shù)據(jù)分成3組(訓(xùn)練、驗(yàn)證和測(cè)試)？

Python 機(jī)器學(xué)習(xí)

翻閱古今 2019-07-20 10:43:48

如何將數(shù)據(jù)分成3組(訓(xùn)練、驗(yàn)證和測(cè)試)？我有一只熊貓數(shù)據(jù)，我想把它分成3組。我知道用列車試驗(yàn)劈開從…sklearn.cross_validation，可以將數(shù)據(jù)分成兩組(訓(xùn)練和測(cè)試)。但是，我無法找到將數(shù)據(jù)分成三組的解決方案。最好，我想要原始數(shù)據(jù)的索引。我知道解決辦法是用train_test_split兩次調(diào)整指數(shù)。但是，是否有更標(biāo)準(zhǔn)/內(nèi)置的方式將數(shù)據(jù)分割成3組而不是2組？

查看完整描述

3 回答

守著星空守著你

TA貢獻(xiàn)1799條經(jīng)驗(yàn) 獲得超8個(gè)贊

但是，有一種方法可以將數(shù)據(jù)集劃分為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)

反對(duì) 回復(fù) 2019-07-20