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 相关文章推荐
Python接收Gmail新邮件并发送到gtalk的方法
Mar 10 Python
浅谈python socket函数中,send与sendall的区别与使用方法
May 09 Python
使用Python写一个贪吃蛇游戏实例代码
Aug 21 Python
Django中的Model操作表的实现
Jul 24 Python
详解Python下Flask-ApScheduler快速指南
Nov 04 Python
Python3.8中使用f-strings调试
May 22 Python
解决pycharm remote deployment 配置的问题
Jun 27 Python
python中class的定义及使用教程
Sep 18 Python
Pytorch训练过程出现nan的解决方式
Jan 02 Python
Python基于进程池实现多进程过程解析
Apr 30 Python
python函数中将变量名转换成字符串实例
May 11 Python
浅谈keras通过model.fit_generator训练模型(节省内存)
Jun 17 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与XML的PDF文档生成技术
2006/10/09 PHP
php小型企业库存管理系统的设计与实现代码
2011/05/16 PHP
深入浅析PHP无限极分类的案例教程
2016/05/09 PHP
完美的php分页类
2017/10/24 PHP
php实现等比例压缩图片
2018/07/26 PHP
Javascript调用XML制作连动下拉列表框
2006/06/25 Javascript
js Html结构转字符串形式显示代码
2011/11/15 Javascript
对frameset、frame、iframe的js操作示例代码
2013/08/16 Javascript
javascript实现浏览器窗口传递参数的方法
2014/09/03 Javascript
基于NodeJS的前后端分离的思考与实践(六)Nginx + Node.js + Java 的软件栈部署实践
2014/09/26 NodeJs
javascript实现表格增删改操作实例详解
2015/05/15 Javascript
js操作DOM--添加、删除节点的简单实例
2016/07/08 Javascript
Vue.js常用指令汇总(v-if、v-for等)
2016/11/03 Javascript
jquery实现input框获取焦点的方法
2017/02/06 Javascript
setTimeout函数的神奇使用
2017/02/26 Javascript
详解如何用模块化的方式写vuejs
2017/12/16 Javascript
ajax请求+vue.js渲染+页面加载的示例
2018/02/11 Javascript
4个顶级开源JavaScript图表库
2018/09/29 Javascript
axios封装,使用拦截器统一处理接口,超详细的教程(推荐)
2019/05/02 Javascript
详解vue-cli中使用rem,vue自适应
2019/05/06 Javascript
jQuery中DOM常见操作实例小结
2019/08/01 jQuery
Layui多选只有最后一个值的解决方法
2019/09/02 Javascript
5分钟快速看懂ES6中的反射与代理
2019/12/19 Javascript
JavaScript进阶(四)原型与原型链用法实例分析
2020/05/09 Javascript
一篇文章让你搞懂JavaScript 原型和原型链
2020/11/23 Javascript
Python字符串拼接、截取及替换方法总结分析
2016/04/13 Python
python中文分词,使用结巴分词对python进行分词(实例讲解)
2017/11/14 Python
详解Python3 中的字符串格式化语法
2020/01/15 Python
程序集与命名空间有什么不同
2014/07/25 面试题
高级销售求职信
2014/02/21 职场文书
中餐厅经理岗位职责
2014/04/11 职场文书
数字化校园建设方案
2014/05/03 职场文书
预备党员群众路线教育实践活动思想汇报2014
2014/10/25 职场文书
挂职个人工作总结
2015/03/05 职场文书
python 逐步回归算法
2021/04/06 Python
pytorch锁死在dataloader(训练时卡死)
2021/05/28 Python