函數(shù)原型：

sklearn.model_selection.train_test_split(train_data,train_target,test_size=0.4,

random_state=0,stratify=y_train)

函數(shù)功能：

將原始數(shù)據(jù)按照比例分割為“測試集”和“訓練集”。返回四個數(shù)組。?

參數(shù)說明：

train_data：要劃分的樣本集。

train_target：要劃分的樣本標簽集。

test_size:測試集所占的比例大小。

random_state:隨機數(shù)種子。

隨機數(shù)種子：其實就是該組隨機數(shù)的編號，在需要重復試驗的時候，保證得到一組一樣的隨機數(shù)。比如你每次都填1，其他參數(shù)一樣的情況下你得到的隨機數(shù)組是一樣的。但填0或不填，每次都會不一樣。

stratify:是為了保持split前類的分布。

比如有100個數(shù)據(jù)，80個屬于A類，20個屬于B類。如果train_test_split(... test_size=0.25, stratify = y_all), 那么split之后數(shù)據(jù)如下：?
training: 75個數(shù)據(jù)，其中60個屬于A類，15個屬于B類。?
testing: 25個數(shù)據(jù)，其中20個屬于A類，5個屬于B類。?

再如----stratify =train_target? : 依據(jù)標簽y，按原數(shù)據(jù)y中各類比例，分配給train和test，使得train和test中各類數(shù)據(jù)的比例與原數(shù)據(jù)集一樣。?

函數(shù)的返回值有四個值：

X_train：訓練特征集

X_test：訓練標簽集

y_train：測試特征集

y_test：測試標簽集

eg:

import numpy as np from sklearn.model_selection import train_test_split X=np.arange(10) y=[0,1,1,0,1,1,1,1,0,0,] ##劃分訓練集占比0.8，訓練集占比0.2 隨機數(shù)種子為0 X_train, X_test, y_train, y_test = train_test_split(X,y,test_size=0.2, random_state=0) print("X=",X) print("X_train=",X_train) print("X_test=",X_test) print("y_train=",y_train) print("y_test=",y_test)result: X= [0 1 2 3 4 5 6 7 8 9] X_train= [4 9 1 6 7 3 0 5] X_test= [2 8] y_train= [1, 0, 1, 1, 1, 0, 0, 1] y_test= [1, 0]

---

建議收藏，以便下次查閱方便

總結(jié)

以上是生活随笔為你收集整理的Scikit-learn——train_test_split（）方法的功能和使用的全部內(nèi)容，希望文章能夠幫你解決所遇到的問題。

如果覺得生活随笔網(wǎng)站內(nèi)容還不錯，歡迎將生活随笔推薦給好友。