TensorFlow函數(shù)：tf.estimator.TrainSpec

tf.estimator.TrainSpec函數(shù)

TrainSpec類

定義在：tensorflow/python/estimator/training.py.

train_and_evaluate調(diào)用的“train”部分的配置.

TrainSpec確定訓(xùn)練的輸入數(shù)據(jù)以及持續(xù)時間.可選的鉤子(hook)在不同訓(xùn)練階段運(yùn)行.

屬性

• hooks
  

  字段號2的別名

• input_fn
  

  字段號0的別名

• max_steps
  

  字段號1的別名

方法

__new__

@ staticmethod 
__new__ ( 
    cls , 
    input_fn , 
    max_steps = None , 
    hooks = None 
)

創(chuàng)建一個已經(jīng)經(jīng)過驗證的TrainSpec實例.

參數(shù)：

• input_fn：訓(xùn)練輸入函數(shù)返回一個元祖：features - Tensor或名為Tensor字符串特征的字典,labels - Tensor或帶有標(biāo)簽的Tensor字典.
  
• max_steps：Int.用于訓(xùn)練模型的總步驟的正數(shù).如果為None,則一直訓(xùn)練.訓(xùn)練input_fn預(yù)計不會產(chǎn)生OutOfRangeError或StopIteration異常.
• hooks：在訓(xùn)練過程中對所有workers(包括chief)運(yùn)行的tf.train.SessionRunHook對象進(jìn)行可迭代處理.

返回值：

tf.estimator.TrainSpec函數(shù)返回一個經(jīng)過驗證的TrainSpec對象.

可能引發(fā)的異常：

• ValueError：如果任何輸入?yún)?shù)無效.
• TypeError：如果任何參數(shù)不是預(yù)期的類型.