python 劃分?jǐn)?shù)據(jù)集為訓(xùn)練集和測(cè)試集的方法
sklearn的cross_validation包中含有將數(shù)據(jù)集按照一定的比例,隨機(jī)劃分為訓(xùn)練集和測(cè)試集的函數(shù)train_test_split
from sklearn.cross_validation import train_test_split #x為數(shù)據(jù)集的feature熟悉,y為label. x_train, x_test, y_train, y_test = train_test_split(x, y, test_size = 0.3)
得到的x_train,y_train(x_test,y_test)的index對(duì)應(yīng)的是x,y中被抽取到的序號(hào)。
若train_test_split傳入的是帶有l(wèi)abel的數(shù)據(jù),則如下代碼:
from sklearn.cross_validation import train_test_split #dat為數(shù)據(jù)集,含有feature和label. train, test = train_test_split(dat, test_size = 0.3)
train,test含有feature和label的。
自己寫了一個(gè)函數(shù):
#X:含label的數(shù)據(jù)集:分割成訓(xùn)練集和測(cè)試集 #test_size:測(cè)試集占整個(gè)數(shù)據(jù)集的比例 def trainTestSplit(X,test_size=0.3): X_num=X.shape[0] train_index=range(X_num) test_index=[] test_num=int(X_num*test_size) for i in range(test_num): randomIndex=int(np.random.uniform(0,len(train_index))) test_index.append(train_index[randomIndex]) del train_index[randomIndex] #train,test的index是抽取的數(shù)據(jù)集X的序號(hào) train=X.ix[train_index] test=X.ix[test_index] return train,test
以上這篇python 劃分?jǐn)?shù)據(jù)集為訓(xùn)練集和測(cè)試集的方法就是小編分享給大家的全部?jī)?nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
- 對(duì)sklearn的使用之?dāng)?shù)據(jù)集的拆分與訓(xùn)練詳解(python3.6)
- Python實(shí)現(xiàn)Keras搭建神經(jīng)網(wǎng)絡(luò)訓(xùn)練分類模型教程
- 在python下實(shí)現(xiàn)word2vec詞向量訓(xùn)練與加載實(shí)例
- python使用pandas抽樣訓(xùn)練數(shù)據(jù)中某個(gè)類別實(shí)例
- Python通過(guò)TensorFLow進(jìn)行線性模型訓(xùn)練原理與實(shí)現(xiàn)方法詳解
- Python 實(shí)現(xiàn)訓(xùn)練集、測(cè)試集隨機(jī)劃分
- 使用Python制作一個(gè)打字訓(xùn)練小工具
- Python編寫打字訓(xùn)練小程序
- Python分割訓(xùn)練集和測(cè)試集的方法示例
- python中如何實(shí)現(xiàn)將數(shù)據(jù)分成訓(xùn)練集與測(cè)試集的方法
- python 使用Yolact訓(xùn)練自己的數(shù)據(jù)集
相關(guān)文章
Python求算數(shù)平方根和約數(shù)的方法匯總
這篇文章主要介紹了 Python求算數(shù)平方根和約數(shù)的方法匯總的相關(guān)資料,需要的朋友可以參考下2016-03-03Python數(shù)據(jù)結(jié)構(gòu)與算法之跳表詳解
跳表是帶有附加指針的鏈表,使用這些附加指針可以跳過(guò)一些中間結(jié)點(diǎn),用以快速完成查找、插入和刪除等操作。本節(jié)將詳細(xì)介紹跳表的相關(guān)概念及其具體實(shí)現(xiàn),需要的可以參考一下2022-02-02windows及l(fā)inux環(huán)境下永久修改pip鏡像源的方法
不知道有沒(méi)有人跟我一樣,在剛接觸Linux時(shí)被系統(tǒng)更新源問(wèn)題搞得暈頭轉(zhuǎn)向,不同的Linux更新源配置也是不一樣的,另外由于默認(rèn)安裝時(shí)的源大都是外國(guó)的更新源,速度相對(duì)國(guó)內(nèi)會(huì)慢很多,接下來(lái)本文主要介紹在windows和linux兩種系統(tǒng)環(huán)境中更新系統(tǒng)源的方法。2016-11-11Python3利用SMTP協(xié)議發(fā)送E-mail電子郵件的方法
SMTP(Simple Mail Transfer Protocol)即簡(jiǎn)單郵件傳輸協(xié)議,它是一組用于由源地址到目的地址傳送郵件的規(guī)則,由它來(lái)控制信件的中轉(zhuǎn)方式。下面這篇文章主要給大家介紹了關(guān)于Python3如何利用SMTP協(xié)議發(fā)送E-mail電子郵件的方法,需要的朋友可以參考下。2017-09-09python機(jī)器學(xué)習(xí)darts時(shí)間序列預(yù)測(cè)和分析
這篇文章主要介紹了python機(jī)器學(xué)習(xí)darts時(shí)間序列預(yù)測(cè)和分析使用實(shí)例探索,有需要的朋友可以借鑒參考下,希望能夠有所幫助,祝大家多多進(jìn)步,早日升職加薪2024-01-01