Python分割訓練集和測試集的方法示例

更新時間：2019年09月19日 10:30:50 作者：積微成著

這篇文章主要介紹了Python分割訓練集和測試集的方法示例,文中通過示例代碼介紹的非常詳細，對大家的學習或者工作具有一定的參考學習價值，需要的朋友們下面隨著小編來一起學習學習吧

數(shù)據(jù)集介紹

使用數(shù)據(jù)集Wine，來自UCI 。包括178條樣本，13個特征。

import pandas as pd
import numpy as np

df_wine = pd.read_csv('https://archive.ics.uci.edu/ml/machine-learning-databases/wine/wine.data', header=None)
df_wine.columns = ['Class label', 'Alcohol',
              'Malic acid', 'Ash',
              'Alcalinity of ash', 'Magnesium',
              'Total phenols', 'Flavanoids',
              'Nonflavanoid phenols',
              'Proanthocyanins',
              'Color intensity', 'Hue',
              'OD280/OD315 of diluted wines',
              'Proline']

分割訓練集和測試集

隨機分割

分為訓練集和測試集

方法：使用scikit-learn中model_selection子模塊的train_test_split函數(shù)

from sklearn.model_selection import train_test_split

X, y = df_wine.ix[:, 1:].values, df_wine.ix[:, 0].values
X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25, random_state=0)#隨機選擇25%作為測試集，剩余作為訓練集

以上就是本文的全部內(nèi)容，希望對大家的學習有所幫助，也希望大家多多支持腳本之家。

您可能感興趣的文章:

python 刪除非空文件夾的實例
下面小編就為大家分享一篇python 刪除非空文件夾的實例，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2018-04-04
Python基礎學習之深淺拷貝問題及遞歸函數(shù)練習
在實際工作中，經(jīng)常涉及到數(shù)據(jù)的傳遞。這篇文章主要為大家介紹了Python的一些基礎學習：深拷貝與淺拷貝問題、遞歸函數(shù)的練習，需要的朋友可以參考一下
2021-12-12
Python中利用原始套接字進行網(wǎng)絡編程的示例
這篇文章主要介紹了Python中利用原始套接字進行網(wǎng)絡編程的示例,使用sock_raw接受和發(fā)送數(shù)據(jù)包可以避開網(wǎng)絡協(xié)議的諸多限制,需要的朋友可以參考下
2015-05-05
基于python的matplotlib制作雙Y軸圖
這篇文章主要介紹了基于python的matplotlib制作雙Y軸圖,文中有非常詳細的代碼示例,對正在學習python的小伙伴們有很好地幫助,需要的朋友可以參考下
2021-04-04
Gradio機器學習模型快速部署工具應用分享前篇
這篇文章主要為大家介紹了Gradio機器學習模型快速部署工具應用分享前篇，有需要的朋友可以借鑒參考下，希望能夠有所幫助，祝大家多多進步，早日升職加薪
2023-04-04
利用python程序幫大家清理windows垃圾
Python 是一種面向?qū)ο?、解釋型計算機程序設計語言，下面這篇文章主要給大家介紹了利用python程序如何實現(xiàn)清理垃圾的功能,本文只適用于windows，有需要的朋友可以參考借鑒，下面來一起學習學習吧。
2017-01-01
pygraphviz安裝教程
Graphviz 是一款由 AT&T Research 和 Lucent Bell 實驗室開源的可視化圖形工具，可以很方便的用來繪制結構化的圖形網(wǎng)絡，支持多種格式輸出，這篇文章主要介紹了pygraphviz安裝教程,需要的朋友可以參考下
2023-02-02
Flask和Django框架中自定義模型類的表名、父類相關問題分析
這篇文章主要介紹了Flask和Django框架中自定義模型類的表名、父類相關問題,結合實例形式對比分析了Flask框架與Django框架表名定義方式的不同之處,并簡單描述了框架的父類繼承問題,需要的朋友可以參考下
2018-07-07
對pyqt5中QTabWidget的相關操作詳解
今天小編就為大家分享一篇對pyqt5中QTabWidget的相關操作詳解，具有很好的參考價值，希望對大家有所幫助。一起跟隨小編過來看看吧
2019-06-06
卷積神經(jīng)網(wǎng)絡的發(fā)展及各模型的優(yōu)缺點及說明
這篇文章主要介紹了卷積神經(jīng)網(wǎng)絡的發(fā)展及各模型的優(yōu)缺點及說明，具有很好的參考價值，希望對大家有所幫助。如有錯誤或未考慮完全的地方，望不吝賜教
2023-02-02