TensorFlow函數(shù):tf.losses.hinge_loss

2018-08-23 11:45 更新

tf.losses.hinge_loss函數(shù)

tf.losses.hinge_loss(
    labels,
    logits,
    weights=1.0,
    scope=None,
    loss_collection=tf.GraphKeys.LOSSES,
    reduction=Reduction.SUM_BY_NONZERO_WEIGHTS
)

定義在:tensorflow/python/ops/losses/losses_impl.py.

在訓練過程中增加了hinge loss.

參數(shù):

  • labels:真實輸出張量,它的形狀應該與logits的形狀相匹配;張量的值預計為0.0或1.0.
  • logits:logits,一個浮動張量.
  • weights:可選的Tensor,其秩為0或與labels的秩相同,并且必須可廣播到labels(即,所有維度必須1,與相應的losses維度相同).
  • scope:計算loss時執(zhí)行的操作范圍.
  • loss_collection:將添加loss的集合.
  • reduction:適用于loss的減少類型.

返回:

加權損失浮動Tensor.如果reduction是NONE,則它的形狀與labels相同;否則,它是標量.

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

  • ValueError:如果logits和labels的形狀不匹配,或者如果labels或logits是None.
以上內(nèi)容是否對您有幫助:
在線筆記
App下載
App下載

掃描二維碼

下載編程獅App

公眾號
微信公眾號

編程獅公眾號