当前位置:首页 >> 脚本专栏

TensorFlow实现从txt文件读取数据

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

以上这篇TensorFlow实现从txt文件读取数据就是小编分享给大家的全部内容了,希望能给大家一个参考,也希望大家多多支持。