使用train_test_split函數可以將原始數據集按照一定比例劃分訓練集和測試集對模型進行訓練

成都創新互聯專注于磁縣企業網站建設,響應式網站建設,商城網站建設。磁縣網站建設公司,為磁縣等地區提供建站服務。全流程定制網站，專業設計，全程項目跟蹤，成都創新互聯專業和態度為您提供的服務

一、舉例

import numpy as np #科學計算庫

from sklearn.model_selection import train_test_split #train_test_split函數

x = np.arange(15).reshape(-1, 3) #生成5行3列的一個矩陣

>>x

array([[ 0,  1,  2],
       [ 3,  4,  5],
       [ 6,  7,  8],
       [ 9, 10, 11],
       [12, 13, 14]])

y = np.arange(5) #5個數的向量

>>y

array([0, 1, 2, 3, 4])

x_train, x_test, y_train, y_test = train_test_split(x, y, test_size=0.2, random_state=1)

>>x_train

array([[ 3,  4,  5],
       [12, 13, 14],
       [ 0,  1,  2],
       [ 9, 10, 11]])

>>x_test

array([[6, 7, 8]])

>>y_train

array([1, 4, 0, 3])

>>y_test

array([2])

二 說明

x，y是原始的數據集。x_train,y_train 是原始數據集劃分出來作為訓練模型的，fit模型的時候用。

x_test,y_test 這部分的數據不參與模型的訓練，而是用于評價訓練出來的模型好壞，score評分的時候用。
test_size=0.2 測試集的劃分比例
random_state=1 隨機種子，如果隨機種子一樣，則隨機生成的數據集是相同的


三 使用KNN

from sklearn.neighbors import KNeighborsClassifier
knn_clf = KNeighborsClassifier()
knn_clf.fit(x_train, y_train) #用fit訓練模型，x_train, y_train是第一步劃分的數據集。

knn_clf.score(x_test, y_test) #score測試模型，x_test, y_test是第一步劃分得到的

另外有需要云服務器可以了解下創新互聯scvps.cn，海內外云服務器15元起步，三天無理由+7*72小時售后在線，公司持有idc許可證，提供“云服務器、裸金屬服務器、高防服務器、香港服務器、美國服務器、虛擬主機、免備案服務器”等云主機租用服務以及企業上云的綜合解決方案，具有“安全穩定、簡單易用、服務可用性高、性價比高”等特點與優勢，專為企業上云打造定制，能夠滿足用戶豐富、多元化的應用場景需求。

網頁標題：機器學習sklearn中的train_test_split()函數-創新互聯
分享地址：http://www.2m8n56k.cn/article38/dhiepp.html

成都網站建設公司_創新互聯，為您提供虛擬主機、企業建站、移動網站建設、靜態網站、ChatGPT、外貿建站

廣告

聲明：本網站發布的內容（圖片、視頻和文字）以用戶投稿、用戶轉載內容為主，如果涉及侵權請盡快告知，我們將會在第一時間刪除。文章觀點不代表本網站立場，如需處理請聯系客服。電話：028-86922220；郵箱：[email protected]。內容未經允許不得轉載，或轉載時需注明來源： 創新互聯