1. 程式人生 > >tf-tips:InTopK和TopK,decay_steps

tf-tips:InTopK和TopK,decay_steps

——InTopK返回一個bool型的tensor,k一般取值為1,表示預測結果(以概率形式表達)中概率最大的是否為target,如果k=3,則是看概率最大的三種結果裡有沒有target,有的化則為True。


——TopK返回型別為tuple:(values, indices)。其中values和indices都是tensor

values(型別與input相同):batch_size*k,  每一張圖片的預測結果包含前k個概率最大的target的概率值。     

indices(int32): batch_size*k,對應相應概率值對應的target的種類或者編號。


——

lr = tf.train.exponential_decay(INITIAL_LEARNING_RATE,
                                global_step,
                                decay_steps,
                                LEARNING_RATE_DECAY_FACTOR,
                                staircase=True)

該函式中的輸入引數decay_steps是學習率每次衰減間隔的訓練步數,LEARNING_RATE_DECAY_FACTOR是衰減係數