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保持一致。