Python 實現(xiàn)訓練集、測試集隨機劃分

更新時間：2020年01月08日 14:52:38 作者：no_error_no_warn

今天小編就為大家分享一篇Python 實現(xiàn)訓練集、測試集隨機劃分，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧

隨機從列表中取出元素：

import random
dataSet = [[0], [1], [2], [3], [4], [5], [6], [7], [8], [9], [10]]
trainDataSet = random.sample(dataSet, 3)

以下函數(shù)，使用于我最近的一個機器學習的項目，將數(shù)據(jù)集數(shù)據(jù)按照比例隨機劃分成訓練集數(shù)據(jù)和測試集數(shù)據(jù)：

import csv
import random
def getDataSet(proportion):
  """
    :exception
      獲取訓練集和測試集(將數(shù)據(jù)按比例隨機劃分)
    :parameter
      proportion - 測試集/數(shù)據(jù)集
    :return
      trainDataSet - 訓練集
      testDataSet - 測試集
    author
      肖政宇
    modify
      2019年5月10日
  """
  dataSet = open('數(shù)據(jù)集.csv')
  dataSetReader = csv.reader(dataSet)
  """
    :exception
      將數(shù)據(jù)保存到數(shù)組
  """
  dataSet = []
  next(dataSetReader, 'none') # 跳過表頭
  data = next(dataSetReader, 'none')
  while (data != 'none'):
    dataSet.append(data)
    data = next(dataSetReader, 'none')
  """
    :exception
      按照比例隨機劃分出訓練集和測試集
  """
  dataNumber = dataSet.__len__() # 數(shù)據(jù)集數(shù)據(jù)條數(shù)
  testNumber = int(dataNumber * proportion) # 測試集數(shù)據(jù)條數(shù)
  testDataSet = [] # 測試數(shù)據(jù)集
  trainDataSet = [] # 訓練數(shù)據(jù)集
 
  testDataSet = random.sample(dataSet, testNumber) # 測試集
  for testData in testDataSet: # 將已經(jīng)選定的測試集數(shù)據(jù)從數(shù)據(jù)集中刪除
    dataSet.remove(testData)
  trainDataSet = dataSet # 訓練集
 
  return trainDataSet, testDataSet

以上這篇Python 實現(xiàn)訓練集、測試集隨機劃分就是小編分享給大家的全部內容了，希望能給大家一個參考，也希望大家多多支持腳本之家。

您可能感興趣的文章:

舉例詳解Python中的split()函數(shù)的使用方法
這篇文章主要介紹了舉例詳解Python中的split()函數(shù)的使用方法,split()函數(shù)的使用是Python學習當中的基礎知識,通常用于將字符串切片并轉換為列表,需要的朋友可以參考下
2015-04-04
解決python中遇到字典里key值為None的情況,取不出來的問題
今天小編就為大家分享一篇解決python中遇到字典里key值為None的情況,取不出來的問題，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-10-10
pandas取出重復數(shù)據(jù)的方法
今天小編就為大家分享一篇pandas取出重復數(shù)據(jù)的方法，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-07-07
python json 遞歸打印所有json子節(jié)點信息的例子
今天小編就為大家分享一篇python json 遞歸打印所有json子節(jié)點信息的例子，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2020-02-02
Python如何識別銀行卡卡號?
今天給大家?guī)淼氖怯嘘PPython的相關知識,文章圍繞著Python如何識別銀行卡卡號展開,文中有非常詳細的代碼示例及介紹,需要的朋友可以參考下
2021-06-06
Python和OpenCV進行多尺度模板匹配實現(xiàn)
本文將實現(xiàn)如何將標準模板匹配擴展到多尺度，使其可以處理模板和輸入圖像大小不同的匹配。具有一定的參考價值，感興趣的小伙伴們可以參考一下
2021-09-09
Python中l(wèi)oguru日志庫的使用
本文主要介紹了Python中l(wèi)oguru日志庫的使用，文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧
2023-03-03
PyQt5每天必學之像素圖控件QPixmap
這篇文章主要為大家詳細介紹了PyQt5每天必學之像素圖控件QPixmap，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2018-04-04
Python中OpenCV實現(xiàn)查找輪廓的實例
本文將結合實例代碼，介紹 OpenCV 如何查找輪廓、獲取邊界框。具有一定的參考價值，感興趣的小伙伴們可以參考一下
2021-06-06
python實現(xiàn)摳圖給證件照換背景源碼
這篇文章主要為大家詳細介紹了python實現(xiàn)摳圖給證件照換背景源碼，具有一定的參考價值，感興趣的小伙伴們可以參考一下
2019-08-08