import numpy as np
from tensorflow.contrib.learn.python.learn.datasets import base
filename = "SiteBUDP_test_label.csv"
test = base.load_csv_without_header(filename=filename,target_dtype=np.int,features_dtype=np.int)
print(test.data.shape,test.target.shape)
test_data=test.data
test_target=test.target
rand_inx=np.random.choice(len(test_data),size=500)
rand_x = test_data[rand_inx]
rand_y = test_target[rand_inx]
print(rand_x.shape,rand_y.shape)
数据集为CSV文件,有10000行,10列
程序输入如下: