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ù)時(shí)間.可選的鉤子(hook)在不同訓(xùn)練階段運(yùn)行.

屬性

• hooks
  

  字段號(hào)2的別名

• input_fn
  

  字段號(hào)0的別名

• max_steps
  

  字段號(hào)1的別名

方法

__new__

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

創(chuàng)建一個(gè)已經(jīng)經(jīng)過驗(yàn)證的TrainSpec實(shí)例.

參數(shù)：

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

返回值：

tf.estimator.TrainSpec函數(shù)返回一個(gè)經(jīng)過驗(yàn)證的TrainSpec對(duì)象.

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

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