這里涉及到的高級別API主要是使用Estimator類來編寫機器學習的程序,此外你還需要用到一些數據導入的知識。
為什么使用Estimator
Estimator類是定義在tf.estimator.Estimator中的,你可以使用其中已經有的Estimator,叫做預創建的Estimator,也可以自定義Estimator。Estimator已經封裝了訓練(train),評估(evaluate),預測(predict),導出以供使用等方法。
此外,Estimator會為我們提供諸如圖構建、創建session等管道工作,不用我們再做這些重復的工作。它還提供了安全的分布式訓練循環。相比於低級的API,我們可以把大部分的時間和精力放在處理數據、訓練模型、調整參數上面,而不是創建張量、構建圖、使用session運行張量上面。
使用Estimator的步驟
1:需要編寫一個數據輸入的函數input_fn
input_fn是輸入函數,這個函數的作用在於對數據進行預處理,並且在模型train,predict,evaluate的時候給模型送進去數據。所以input_fn主要作用的時機在模型訓練、預測和評估的時候,在模型定義的時候不需要傳入輸入函數,而是傳入一個預定義的特征列。可以使用系統自帶的函數,可以編寫自定義的輸入函數。
使用系統自帶的數據輸入函數:
系統自帶的輸入函數為tf.estimator.inputs.numpy_input_fn,它的輸入參數如下:
def numpy_input_fn(x, y=None, batch_size=128, num_epochs=1, shuffle=None, queue_capacity=1000, num_threads=1)
x為numpy數組或者numpy數組的字典,當為numpy數組的時候,這個數組被當做單一的特征對待。
一個例子如下,這個例子是tf.estimator Quickstart tutorial中的一段代碼:
import numpy as np training_set = tf.contrib.learn.datasets.base.load_csv_with_header( filename=IRIS_TRAINING, target_dtype=np.int, features_dtype=np.float32) train_input_fn = tf.estimator.inputs.numpy_input_fn( x={"x": np.array(training_set.data)}, y=np.array(training_set.target), num_epochs=None, shuffle=True) classifier.train(input_fn=train_input_fn, steps=2000)
自定義導入數據的函數:
要自定義導入函數,要知道tensorflow中關於數據的概念,以及知道自定義的函數應該返回的值,下面我將梳理一下這里面的概念:
自定義函數的基本框架以及返回值
def my_input_fn(): # 在這里進行數據的預處理... # ...返回兩個值 1) 一個由特征列和包含特征的Tensors組成的映射(字典) 2) 一個包含labels的Tensor return feature_cols, labels
自定義函數需要返回兩個值,一個值是feature_cols,是一個字典,其中字典的key為特征的列名稱,字典的value為包含特征值的Tensor對象。labels是一個包含標簽值的Tensor對象。
tf.data.API對於數據的兩個抽象:
使用tf.data.API來構建數據輸入的管道,幫助我們導入數據,無論是圖像,文本還是分布式的數據,都可以用它來完成。
一個抽象的概念是tf.data.Dataset,一個Dataset是一個數據集,它是由一系列的元素組成的,每個元素的類型都是相同的。其中每個元素包含一個或者多個Tensor對象。我們可以以兩種方式來創建Dataset對象,一種方式是創建它的來源,比如使用Dataset.from_tensor_slices(),可以使用張量來創建Dataset對象,另外一種方式是運用轉換的方式,可以將一個Dataset來變成另外一個Dataset,比如Dataset.batch()。
另外一個抽象的概念是tf.data.Iterator,它代表的是迭代器。表示的是如何從數據集里面取出元素,最簡答的迭代器是單次迭代器,Dataset.make_one_shot_iterator()可以創建單次迭代器。創建迭代器以后,可以使用Iterator.get_next()來獲取下一個元素。
其它的創建數據集的方法:
Dataset.from_tensor()創建一個Dataset,並將傳入的Tensor當做一個元素。 Dataset.from_tensor_slices()會創建一個Dataset,並且將傳入的Tensor在第0維上面切面,分成一些列的元素。還可以使用TFRecordDataset來獲得磁盤上面TFRecord格式的數據。
其它的創建迭代器的方法:
除了dataset.make_one_shot_iterator()這種單次迭代器以外,你還可以創建可初始化、可重新初始化、可饋送迭代器。
導入數據集的基本的工作機制:
1:創建Dataset對象 –> 2:將Dataset進行轉化 –> 3:創建迭代器 –> 4:用迭代器返回下一個元素。
下面用一個例子來說明一下:
from tensorflow.python.data import Dataset import numpy as np def my_input_fn(features, targets, batch_size=1, shuffle=True, num_epochs=None): """自定義的輸入函數 Args: features: 使用pandas中的DataFrame對象來表示的features targets: 使用pandas的taFrame對象表示的targets batch_size: 批次的大小 shuffle: 是否將數據進行重新打亂 num_epochs: 需要重復的epochs的數量,一個epochs代表一個訓練周期. None = repeat indefinitely Returns: 下一批次數據的元組 (features, labels) """ # 將pandas對象轉換為字典,其中字典的值為numpy的數組 features = {key:np.array(value) for key,value in dict(features).items()} # 創建一個Dataset,並且設置好批次和重復的次數 ds = Dataset.from_tensor_slices((features,targets)) # warning: 2GB limit ds = ds.batch(batch_size).repeat(num_epochs) # 是否進行數據擾動 if shuffle: ds = ds.shuffle(10000) # 返回下個批次的數據 features, labels = ds.make_one_shot_iterator().get_next() return features, labels
上面自定義了數據導入的函數,使用Dataset.from_tensor_slices()來創建Dataset。然后使用batch、repeat、shuffle進行轉換。 接着創建迭代器,並且獲得下一個元素。
2:定義特征列
使用tf.feature_column來標識特征名稱、類型和任何輸入預處理。
特征列在原始數據和模型之間起到了連接的作用。在編寫模型的時候需要預先確定輸入數據的特征列。
比如包含經度和維度兩個特征的特征列,它們都是數值類型,這個特征列在模型定義的時候需要傳入:
import tensorflow as tf longitude = tf.feature_column.numeric_column('longitude') latitude = tf.feature_column.numeric_column('latitude') feature_column = [longitude, latitude]
特征列在原始數據與模型所需的數據之間架起了橋梁。
3:實例化相關的預創建的Estimator
這個步驟就簡單了,以深度學習模型為例,運用上面創建的經緯度特征列,使用10*10的隱層創建一個深度神經網絡的回歸模型:
hidden_units = [10, 10] dnn_regressor = tf.estimator.DNNRegressor( feature_columns=feature_columns, hidden_units=hidden_units, )
4:調用訓練、評估或推理方法
使用上述創建的模型進行train、evaluate、predict操作。首先需要定理訓練的輸入函數,將訓練集的特征和標簽都傳進去,然后開始訓練,例子如下:
training_input_fn = lambda:my_input_fn(train_df, train_target_df) dnn_regressor.train( input_fn=training_input_fn, steps=300 )
參考:
Estimator 高級的API,介紹了創建estimator的流程
導入數據 介紹了數據集,還有迭代器的知識
Building Input Functions with tf.estimator 講解了如何定義輸入函數
特征列 詳細介紹了特征列,里面有9中特征列可以學習
google機器學習速成課程的神經網絡簡介 ,完整的機器學習過程