隨機(jī)從列表中取出元素：

import random

dataSet = [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10]]

trainDataSet = random.sample(dataSet, 3)

以下函數(shù)，使用于我最近的一個(gè)機(jī)器學(xué)習(xí)的項(xiàng)目，將數(shù)據(jù)集數(shù)據(jù)按照比例隨機(jī)劃分成訓(xùn)練集數(shù)據(jù)和測(cè)試集數(shù)據(jù)：

import csv

import random

def getDataSet(proportion):

"""

:exception

獲取訓(xùn)練集和測(cè)試集(將數(shù)據(jù)按比例隨機(jī)劃分)

:parameter

proportion - 測(cè)試集/數(shù)據(jù)集

:return

trainDataSet - 訓(xùn)練集

testDataSet - 測(cè)試集

author

肖政宇

modify

2019年5月10日

"""

dataSet = open('數(shù)據(jù)集.csv')

dataSetReader = csv.reader(dataSet)

"""

:exception

將數(shù)據(jù)保存到數(shù)組

"""

dataSet = []

next(dataSetReader, 'none') # 跳過(guò)表頭

data = next(dataSetReader, 'none')

while (data != 'none'):

dataSet.append(data)

data = next(dataSetReader, 'none')

"""

:exception

按照比例隨機(jī)劃分出訓(xùn)練集和測(cè)試集

"""

dataNumber = dataSet.__len__() # 數(shù)據(jù)集數(shù)據(jù)條數(shù)

testNumber = int(dataNumber * proportion) # 測(cè)試集數(shù)據(jù)條數(shù)

testDataSet = [] # 測(cè)試數(shù)據(jù)集

trainDataSet = [] # 訓(xùn)練數(shù)據(jù)集

testDataSet = random.sample(dataSet, testNumber) # 測(cè)試集

for testData in testDataSet: # 將已經(jīng)選定的測(cè)試集數(shù)據(jù)從數(shù)據(jù)集中刪除

dataSet.remove(testData)

trainDataSet = dataSet # 訓(xùn)練集

return trainDataSet, testDataSet

以上這篇Python 實(shí)現(xiàn)訓(xùn)練集、測(cè)試集隨機(jī)劃分就是小編分享給大家的全部?jī)?nèi)容了，希望能給大家一個(gè)參考，也希望大家多多支持python博客。

總結(jié)

以上是生活随笔為你收集整理的python 识别数字字母 训练集_Python 实现训练集、测试集随机划分的全部?jī)?nèi)容，希望文章能夠幫你解決所遇到的問(wèn)題。

如果覺(jué)得生活随笔網(wǎng)站內(nèi)容還不錯(cuò)，歡迎將生活随笔推薦給好友。