Python分割训练集和测试集的方法示例
更新时间：2019年09月19日 10:30:50 作者：积微成著
这篇文章主要介绍了Python分割训练集和测试集的方法示例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧
数据集介绍
使用数据集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函数
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%作为测试集，剩余作为训练集
以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持脚本之家。
相关文章
下面小编就为大家分享一篇python 删除非空文件夹的实例，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2018-04-04
在实际工作中，经常涉及到数据的传递。这篇文章主要为大家介绍了Python的一些基础学习：深拷贝与浅拷贝问题、递归函数的练习，需要的朋友可以参考一下2021-12-12
这篇文章主要介绍了Python中利用原始套接字进行网络编程的示例,使用sock_raw接受和发送数据包可以避开网络协议的诸多限制,需要的朋友可以参考下2015-05-05
这篇文章主要介绍了基于python的matplotlib制作双Y轴图,文中有非常详细的代码示例,对正在学习python的小伙伴们有很好地帮助,需要的朋友可以参考下2021-04-04
这篇文章主要为大家介绍了Gradio机器学习模型快速部署工具应用分享前篇，有需要的朋友可以借鉴参考下，希望能够有所帮助，祝大家多多进步，早日升职加薪2023-04-04
Python 是一种面向对象、解释型计算机程序设计语言，下面这篇文章主要给大家介绍了利用python程序如何实现清理垃圾的功能,本文只适用于windows，有需要的朋友可以参考借鉴，下面来一起学习学习吧。2017-01-01
Graphviz 是一款由 AT&T Research 和 Lucent Bell 实验室开源的可视化图形工具，可以很方便的用来绘制结构化的图形网络，支持多种格式输出，这篇文章主要介绍了pygraphviz安装教程,需要的朋友可以参考下2023-02-02
Flask和Django框架中自定义模型类的表名、父类相关问题分析
这篇文章主要介绍了Flask和Django框架中自定义模型类的表名、父类相关问题,结合实例形式对比分析了Flask框架与Django框架表名定义方式的不同之处,并简单描述了框架的父类继承问题,需要的朋友可以参考下2018-07-07
今天小编就为大家分享一篇对pyqt5中QTabWidget的相关操作详解，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-06-06
这篇文章主要介绍了卷积神经网络的发展及各模型的优缺点及说明，具有很好的参考价值，希望对大家有所帮助。如有错误或未考虑完全的地方，望不吝赐教2023-02-02
最新评论