博客
关于我
强烈建议你试试无所不能的chatGPT,快点击我
python中如何实现将数据分成训练集与测试集
阅读量:7040 次
发布时间:2019-06-28

本文共 1390 字,大约阅读时间需要 4 分钟。

接下来,直接给出大家响应的代码,并对每一行进行标注,希望能够帮到大家。

需要用到的是库是。numpy 、sklearn。

#导入相应的库(对数据库进行切分需要用到的库是sklearn.model_selection 中的 train_test_split)import numpy as npfrom sklearn.model_selection import train_test_split #首先,读取.CSV文件成矩阵的形式。my_matrix = np.loadtxt(open("xxxxxx.csv"),delimiter=",",skiprows=0) #对于矩阵而言,将矩阵倒数第一列之前的数值给了X(输入数据),将矩阵大最后一列的数值给了y(标签)X, y = my_matrix[:,:-1],my_matrix[:,-1] #利用train_test_split方法,将X,y随机划分问,训练集(X_train),训练集标签(X_test),测试卷(y_train), 测试集标签(y_test),安训练集:测试集=7:3的 概率划分,到此步骤,可以直接对数据进行处理X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42) #此步骤,是为了将训练集与数据集的数据分别保存为CSV文件 #np.column_stack将两个矩阵进行组合连接train= np.column_stack((X_train,y_train)) #numpy.savetxt 将txt文件保存为。csv结尾的文件numpy.savetxt('train_usual.csv',train, delimiter = ',')test = np.column_stack((X_test, y_test))numpy.savetxt('test_usual.csv', test, delimiter = ',')

完整没解释的代码部分为

import numpy as npfrom sklearn.model_selection import train_test_splitmy_matrix = np.loadtxt(open("xxxxx.csv"),delimiter=",",skiprows=0)X, y = my_matrix[:,:-1],my_matrix[:,-1]X_train, X_test, y_train, y_test = train_test_split(X, y, test_size=0.3, random_state=42)train= np.column_stack((X_train,y_train))numpy.savetxt('train_usual.csv',train, delimiter = ',')test = np.column_stack((X_test, y_test))numpy.savetxt('test_usual.csv', test, delimiter = ',')

转载于:https://www.cnblogs.com/DeepRunning/p/9205887.html

你可能感兴趣的文章
正则表达式
查看>>
bind this指针
查看>>
paper 135:关于C#泛型的知识点
查看>>
第二十四条:消除非受检警告
查看>>
给阅读的网页作标记
查看>>
vue条件渲染
查看>>
转 MySQL数据库基础
查看>>
Oracle dblink创建
查看>>
python04 while循环
查看>>
web 开发之酷炫--- 酷炫展示
查看>>
ubuntu 解压命令全部
查看>>
Chrome教程(一)NetWork面板分析网络请求
查看>>
第十八回  基础才是重中之重~开发人员应学会用throw
查看>>
Rosenblatt's perceptron
查看>>
1570:基础练习 分解质因数
查看>>
判断ie浏览器7、8、9三个版本
查看>>
GDUFE ACM-1124
查看>>
Schwarz积分公式
查看>>
工作中常用的 Linux 命令
查看>>
English Corner
查看>>