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,这门以简洁、易学、强大著称的编程语言,更是成为无数开发者的大宝剑,今天,我们要探讨的就是如何用 Python 远程控制 Windows 服务器,需要的朋友可以参考下2024-07-07
本文主要介绍了Pandas统计计数value_counts()的使用，文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值，需要的朋友们下面随着小编来一起学习学习吧2022-07-07
这篇文章主要为大家详细介绍了如何利用Python语言实现yaml与json文件的批量互转，文中的示例代码讲解详细，感兴趣的小伙伴可以动手尝试一下2022-07-07
这篇文章主要介绍了基于python实现图片转字符画代码实例,文中通过示例代码介绍的非常详细，对大家的学习或者工作具有一定的参考学习价值,需要的朋友可以参考下2020-09-09
进程在创建之后是没有办法获取返回值的，但有的时候两个进程之间需要进行相互之间的配合才能完成工作，这就需要通信的帮助。本文主要介绍了Python中进程间的通信方式，需要的可以了解一下2022-04-04
今天小编就为大家分享一篇在Python 不同级目录之间模块的调用方法，具有很好的参考价值，希望对大家有所帮助。一起跟随小编过来看看吧2019-01-01
本文介绍了将PIL格式、数组和图片转换为二进制的不同方法,包括使用PIL库、OpenCV和直接读取二进制,此外,还提到了数据传输中base64格式的应用,这些信息对需要进行图片数据处理和转换的开发者非常有用2024-09-09
这篇文章主要介绍了python提取内容关键词的方法,适用于英文关键词的提取,非常具有实用价值,需要的朋友可以参考下2015-03-03
这篇文章主要介绍了python删除过期log文件，分享了相关代码示例，小编觉得还是挺不错的，具有一定借鉴价值，需要的朋友可以参考下2018-01-01
这篇文章主要介绍了Python字符串拼接六种方法介绍，具有一定借鉴价值，需要的朋友看可以参考下。2017-12-12
最新评论