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

2019-11-25 11:41:36

字体：大中小

来源：转载

供稿：网友

数据集介绍

使用数据集Wine，来自UCI 。包括178条样本，13个特征。

import pandas as pdimport numpy as npdf_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_splitX, y = df_wine.ix[:, 1:].values, df_wine.ix[:, 0].valuesX_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.25, random_state=0)#随机选择25%作为测试集，剩余作为训练集

以上就是本文的全部内容，希望对大家的学习有所帮助，也希望大家多多支持武林网。

上一篇：基于python的selenium两种文件上传操作实现详解

下一篇：python Dijkstra算法实现最短路径问题的方法