2016-07-27 52 views
1

Tensorflow中是否實施了「二元邏輯損失」? 我發現下面的定義實現它:Tensorflow中的二元Logistic損失

loss(x, y) = log(1 + exp(-x))  if y == 1 
      log(1 + exp(-x)) + x if y == 0 

我想檢查它是否做到這一點的最好辦法,如果它不存在:

loss = tf.reduce_mean(tf.log(1 + tf.exp(-x)) + tf.abs(y - 1) *x) 

回答

0

退房tf.equal()(返回一個布爾值),然後將生成的布爾值輸入到tf.select()

相關問題