TensorFlow从txt文件中读取数据的方法很多有种,我比较常用的是下面两种:
【1】np.loadtxt
import numpy as np data=np.loadtxt('ex1data1.txt',dtype='float',delimiter=',') X_train=data[:,0] y_train=data[:,1]
【2】pd.read_csv
import pandas as pd data=pd.read_csv("ex2data2.txt",names=['x1','x2','y']) data=np.matrix(data) x_train=data[:,:2] y_train=data[:,2]
Ps:注意读取出来的数据的shape,要与你定义的placeholder的shape保持一致。