Python數(shù)據(jù)集切分實(shí)例
在處理數(shù)據(jù)過程中經(jīng)常要把數(shù)據(jù)集切分為訓(xùn)練集和測試集,因此記錄一下切分代碼。
''' data:數(shù)據(jù)集 test_ratio:測試機(jī)占比 如果data為numpy.numpy.ndarray直接使用此代碼 如果data為pandas.DatFrame類型則 return data[train_indices],data[test_indices] 修改為 return data.iloc[train_indices],data.iloc[test_indices] ''' def split_train(data,test_ratio): shuffled_indices=np.random.permutation(len(data)) test_set_size=int(len(data)*test_ratio) test_indices =shuffled_indices[:test_set_size] train_indices=shuffled_indices[test_set_size:] return data[train_indices],data[test_indices]
測試代碼如下:
import numpy as np import pandas as pd data=np.random.randint(100,size=[25,4]) print(data)
結(jié)果如下:
從上圖可以看出,原數(shù)據(jù)集按照5:1被隨機(jī)分為兩部分。但是此種方法存在一個(gè)缺點(diǎn)–每次調(diào)用次函數(shù)切分同一個(gè)數(shù)據(jù)集切分出來的結(jié)果都不一樣,因此常在np.random.permutation(len(data))先調(diào)用np.random.seed(int)函數(shù),來確保每次切分來的結(jié)果相同。
因此將上述函數(shù)改為:
def split_train(data,test_ratio): np.random.seed(43) shuffled_indices=np.random.permutation(len(data)) test_set_size=int(len(data)*test_ratio) test_indices =shuffled_indices[:test_set_size] train_indices=shuffled_indices[test_set_size:] return data[train_indices],data[test_indices]
這個(gè)函數(shù)np.random.seed(43)當(dāng)參數(shù)為同一整數(shù)時(shí)產(chǎn)生的隨機(jī)數(shù)相同。
以上這篇Python數(shù)據(jù)集切分實(shí)例就是小編分享給大家的全部內(nèi)容了,希望能給大家一個(gè)參考,也希望大家多多支持腳本之家。
相關(guān)文章
python torch.utils.data.DataLoader使用方法
這篇文章主要介紹了python torch.utils.data.DataLoader使用方法,文中通過示例代碼介紹的非常詳細(xì),對大家的學(xué)習(xí)或者工作具有一定的參考學(xué)習(xí)價(jià)值,需要的朋友們下面隨著小編來一起學(xué)習(xí)學(xué)習(xí)吧2020-04-04python json 遞歸打印所有json子節(jié)點(diǎn)信息的例子
今天小編就為大家分享一篇python json 遞歸打印所有json子節(jié)點(diǎn)信息的例子,具有很好的參考價(jià)值,希望對大家有所幫助。一起跟隨小編過來看看吧2020-02-02Python實(shí)現(xiàn)優(yōu)雅編寫LaTeX的示例代碼
LaTeX?是一種廣泛用于排版學(xué)術(shù)論文、報(bào)告、書籍和演示文稿的標(biāo)記語言,本文主要為大家詳細(xì)介紹了如何使用?Python?來優(yōu)雅地編寫?LaTeX,提高效率并減少錯(cuò)誤,需要的可以參考下2024-02-02Python程序?qū)崿F(xiàn)向MySQL存放圖片
這篇文章主要介紹了Python程序?qū)崿F(xiàn)向MySQL存放圖片,具有很好的參考價(jià)值,希望對大家有所幫助。如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2023-03-03Python之多線程退出與停止的一種實(shí)現(xiàn)思路
這篇文章主要介紹了Python之多線程退出與停止的一種實(shí)現(xiàn)思路,具有很好的參考價(jià)值,希望對大家有所幫助,如有錯(cuò)誤或未考慮完全的地方,望不吝賜教2024-02-02