TensorFlow 2 version | View source on GitHub |
Computes the categorical hinge loss between y_true
and y_pred
.
tf.keras.losses.categorical_hinge(
y_true, y_pred
)
Args | |
---|---|
y_true
|
The ground truth values. y_true values are expected to be -1 or 1.
If binary (0 or 1) labels are provided they will be converted to -1 or 1.
|
y_pred
|
The predicted values. |
Returns | |
---|---|
A tensor. |