提取训练，测试集

最新推荐文章于 2024-07-12 16:16:27 发布

bombSKLK

最新推荐文章于 2024-07-12 16:16:27 发布

阅读量760

点赞数

分类专栏：机器学习（python）文章标签： python

本文链接：https://blog.csdn.net/bombSKLK/article/details/50417670

版权

机器学习（python）专栏收录该内容

2 篇文章 0 订阅

订阅专栏

只是一个小代码

import numpy as np
from numpy import *
import pandas as pd
from pandas import DataFrame

df = pd.read_csv('train_x.csv')
trainingSet = np.array(df)

labels = pd.read_csv('train_y.csv')
labelsArr = np.array(labels)

trainMat = [];trainLabels = []
testMat = [];testLabels = []

for i in range(0,14999,1):
    
    if i%10 == 1:
        testMat.append(trainingSet[i,:])
        testLabels.append(labelsArr[i,:])
    else:
        trainMat.append(trainingSet[i,:])
        trainLabels.append(labelsArr[i,:])
	
 #验证
shape(testMat)   #(1500L,1139L)
shape(trainMat)  #(13499L,1139L)