函數原型：

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

random_state=0,stratify=y_train)

函數功能：

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

參數說明：

train_data：要劃分的樣本集。

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

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

random_state:隨機數種子。

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

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

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

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

函數的返回值有四個值：

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 隨機數種子為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]

---

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

總結

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

如果覺得生活随笔網站內容還不錯，歡迎將生活随笔推薦給好友。