Sklearn庫的train_test_split()

作用:將數據集隨機劃分爲訓練集和測試集

使用:X_train,X_test,Y_train,Y_test = train_test_split(X,Y,random_state = 0)

返回:X_train,X_test,Y_train,Y_test

參數:
X:原始數據集中的樣品特徵
Y:原始數據集中的預測結果
random_state:該組隨機數的編號,重複劃分時一樣的數值保證得到一樣的隨機數,不填時不能保證

———關注我的公衆號,一起學數據挖掘————
這裏寫圖片描述

發佈了29 篇原創文章 · 獲贊 25 · 訪問量 5萬+
發表評論
所有評論
還沒有人評論,想成為第一個評論的人麼? 請在上方評論欄輸入並且點擊發布.
相關文章