首頁猿問如何將數據分成3組(訓練、驗證和測試)？

如何將數據分成3組(訓練、驗證和測試)？

Python 機器學習

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

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

查看完整描述

3 回答

守著星空守著你

TA貢獻1799條經驗獲得超8個贊

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

反對回復 2019-07-20