1. 程式人生 > 程式設計 >TensorFlow實現從txt檔案讀取資料

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檔案讀取資料就是小編分享給大家的全部內容了,希望能給大家一個參考,也希望大家多多支援我們。