python 劃分數(shù)據(jù)集為訓練集和測試集的方法
sklearn的cross_validation包中含有將數(shù)據(jù)集按照一定的比例,隨機劃分為訓練集和測試集的函數(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對應的是x,y中被抽取到的序號。
若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的。
自己寫了一個函數(shù):
#X:含label的數(shù)據(jù)集:分割成訓練集和測試集 #test_size:測試集占整個數(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的序號 train=X.ix[train_index] test=X.ix[test_index] return train,test
以上這篇python 劃分數(shù)據(jù)集為訓練集和測試集的方法就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支持腳本之家。
- 對sklearn的使用之數(shù)據(jù)集的拆分與訓練詳解(python3.6)
- Python實現(xiàn)Keras搭建神經網絡訓練分類模型教程
- 在python下實現(xiàn)word2vec詞向量訓練與加載實例
- python使用pandas抽樣訓練數(shù)據(jù)中某個類別實例
- Python通過TensorFLow進行線性模型訓練原理與實現(xiàn)方法詳解
- Python 實現(xiàn)訓練集、測試集隨機劃分
- 使用Python制作一個打字訓練小工具
- Python編寫打字訓練小程序
- Python分割訓練集和測試集的方法示例
- python中如何實現(xiàn)將數(shù)據(jù)分成訓練集與測試集的方法
- python 使用Yolact訓練自己的數(shù)據(jù)集
相關文章
Python求算數(shù)平方根和約數(shù)的方法匯總
這篇文章主要介紹了 Python求算數(shù)平方根和約數(shù)的方法匯總的相關資料,需要的朋友可以參考下2016-03-03
windows及l(fā)inux環(huán)境下永久修改pip鏡像源的方法
不知道有沒有人跟我一樣,在剛接觸Linux時被系統(tǒng)更新源問題搞得暈頭轉向,不同的Linux更新源配置也是不一樣的,另外由于默認安裝時的源大都是外國的更新源,速度相對國內會慢很多,接下來本文主要介紹在windows和linux兩種系統(tǒng)環(huán)境中更新系統(tǒng)源的方法。2016-11-11
Python3利用SMTP協(xié)議發(fā)送E-mail電子郵件的方法
SMTP(Simple Mail Transfer Protocol)即簡單郵件傳輸協(xié)議,它是一組用于由源地址到目的地址傳送郵件的規(guī)則,由它來控制信件的中轉方式。下面這篇文章主要給大家介紹了關于Python3如何利用SMTP協(xié)議發(fā)送E-mail電子郵件的方法,需要的朋友可以參考下。2017-09-09

