mlx.nn.losses.hinge_loss

目录

mlx.nn.losses.hinge_loss#

class hinge_loss(inputs: array, targets: array, reduction: Literal['none', 'mean', 'sum'] = 'none')#

计算输入和目标之间的合页损失。

\[\text{hinge}(y, y_{\text{pred}}) = \max(0, 1 - y \cdot y_{\text{pred}})\]
参数:
  • inputs (array) – 预测值。

  • targets (array) – 目标值。应为 -1 或 1。

  • reduction (str, optional) – 指定应用于输出的归约方式:'none' | 'mean' | 'sum'。默认值:'none'

返回值:

计算出的合页损失。

返回类型:

array