Python分割训练集和测试集的方法示例


Posted in Python onSeptember 19, 2019

数据集介绍

使用数据集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 相关文章推荐
怎么使用pipenv管理你的python项目
Mar 12 Python
对numpy的array和python中自带的list之间相互转化详解
Apr 13 Python
Python实现字典(dict)的迭代操作示例
Jun 05 Python
pyQt4实现俄罗斯方块游戏
Jun 26 Python
Opencv+Python 色彩通道拆分及合并的示例
Dec 08 Python
python中字符串数组逆序排列方法总结
Jun 23 Python
python网络编程之多线程同时接受和发送
Sep 03 Python
python取均匀不重复的随机数方式
Nov 27 Python
使用Tkinter制作信息提示框
Feb 18 Python
Python restful框架接口开发实现
Apr 13 Python
pyspark对Mysql数据库进行读写的实现
Dec 30 Python
Python实现王者荣耀自动刷金币的完整步骤
Jan 22 Python
python Dijkstra算法实现最短路径问题的方法
Sep 19 #Python
PHP统计代码行数的小代码
Sep 19 #Python
Python实现代码统计工具
Sep 19 #Python
python实现统计代码行数的小工具
Sep 19 #Python
python日志模块logbook使用方法
Sep 19 #Python
python统计指定目录内文件的代码行数
Sep 19 #Python
python如何从文件读取数据及解析
Sep 19 #Python
You might like
用PHP 快速生成 Flash 动画的方法
2007/03/06 PHP
PHP mysqli事务操作常用方法分析
2017/07/22 PHP
window.location和document.location的区别分析
2008/12/23 Javascript
firefox插件Firebug的使用教程
2010/01/02 Javascript
js 模拟气泡屏保效果代码
2010/07/10 Javascript
JavaScript自定义DateDiff函数(兼容所有浏览器)
2012/03/01 Javascript
JQuery弹出炫丽对话框的同时让背景变灰色
2014/05/22 Javascript
JavaScript解析json格式数据简单示例
2014/12/09 Javascript
jQuery的观察者模式详解
2014/12/22 Javascript
分享网页检测摇一摇实例代码
2016/01/14 Javascript
浅谈js内置对象Math的属性和方法(推荐)
2016/09/19 Javascript
Bootstrap按钮组实例详解
2017/07/03 Javascript
vue实现文章内容过长点击阅读全文功能的实例
2017/12/28 Javascript
redux中间件之redux-thunk的具体使用
2018/04/17 Javascript
elementUI多选框反选的实现代码
2019/04/03 Javascript
vue+webpack 更换主题N种方案优劣分析
2019/10/28 Javascript
[57:22]完美世界DOTA2联赛PWL S2 FTD vs PXG 第二场 11.27
2020/12/01 DOTA
[32:56]完美世界DOTA2联赛PWL S3 Rebirth vs CPG 第二场 12.11
2020/12/16 DOTA
探究数组排序提升Python程序的循环的运行效率的原因
2015/04/01 Python
浅析Python中的多进程与多线程的使用
2015/04/07 Python
python类继承与子类实例初始化用法分析
2015/04/17 Python
python输出当前目录下index.html文件路径的方法
2015/04/28 Python
Python中super()函数简介及用法分享
2016/07/11 Python
Python中动态创建类实例的方法
2017/03/24 Python
windows下Python实现将pdf文件转化为png格式图片的方法
2017/07/21 Python
python使用Matplotlib绘制分段函数
2018/09/25 Python
python 对key为时间的dict排序方法
2018/10/17 Python
python+opencv 读取文件夹下的所有图像并批量保存ROI的方法
2019/01/10 Python
Python 实现网课实时监控自动签到、打卡功能
2020/03/12 Python
土木工程专业自荐信
2013/10/04 职场文书
统计系教授推荐信
2014/02/28 职场文书
试用期自我鉴定范文
2014/03/20 职场文书
交通事故协议书范文
2014/04/16 职场文书
团代会开幕词
2015/01/28 职场文书
2015年学校综合治理工作总结
2015/07/20 职场文书
AudioContext 实现音频可视化(web技术分享)
2022/02/24 Javascript