關於train_test_split和cross_val_score交叉檢驗


train_test_split分組

  • train_test_split函數用於將矩陣隨機划分為訓練子集和測試子集,並返回划分好的訓練集測試集樣本和訓練集測試集標簽。
  • X_train,X_test, y_train, y_test =cross_validation.train_test_split(train_data,train_target,test_size=0.3, random_state=0)
    • train_data:被划分的樣本特征集
    • train_target:被划分的樣本標簽
    • test_size:如果是浮點數,在0-1之間,表示樣本占比;如果是整數的話就是樣本的數量
    • random_state:是隨機數的種子
from sklearn.model_selection import train_test_split

cross_val_score

from sklearn.model_selection import cross_val_score
scores = cross_val_score(model, X, y, cv=n_folds)
# cv 設置交叉檢驗的次數

得到一個一維數組


免責聲明!

本站轉載的文章為個人學習借鑒使用,本站對版權不負任何法律責任。如果侵犯了您的隱私權益,請聯系本站郵箱yoyou2525@163.com刪除。



 
粵ICP備18138465號   © 2018-2025 CODEPRJ.COM