Python tf.keras.metrics.Hinge用法及代碼示例

計算 y_true 和 y_pred 之間的鉸鏈度量。

繼承自：MeanMetricWrapper、Mean、Metric、Layer、Module

用法

tf.keras.metrics.Hinge(
    name='hinge', dtype=None
)

y_true 值應為 -1 或 1。如果提供二進製(0 或 1)標簽，我們會將它們轉換為 -1 或 1。

單機使用：

m = tf.keras.metrics.Hinge()
m.update_state([[0, 1], [0, 0]], [[0.6, 0.4], [0.4, 0.6]])
m.result().numpy()
1.3

m.reset_state()
m.update_state([[0, 1], [0, 0]], [[0.6, 0.4], [0.4, 0.6]],
               sample_weight=[1, 0])
m.result().numpy()
1.1

compile() API 的用法：

model.compile(optimizer='sgd', loss='mse', metrics=[tf.keras.metrics.Hinge()])