tf.compat.v1.losses.hinge_loss
Добавляет функцию потерь hinge к процедуре обучения.
tf.compat.v1.losses.hinge_loss(
labels, logits, weights=1.0, scope=None, loss_collection=tf.GraphKeys.LOSSES,
reduction=Reduction.SUM_BY_NONZERO_WEIGHTS
)
| Аргументы | |
|---|---|
labels | Тензор правильного выходного значения. Его форма должна совпадать с формой logits. Ожидается, что значения тензора будут 0.0 или 1.0. Внутренне метки {0,1} преобразуются в {-1,1} при вычислении функции потерь hinge. |
logits | Логиты, числовой тензор. Обратите внимание, что предполагается, что логиты неограничены и центрированы на 0. Значение > 0 (соответственно < 0) рассматривается как положительное (соответственно отрицательное) бинарное предсказание. |
weights | Необязательный Tensor, ранг которого равен либо 0, либо совпадает с рангом labels, и он должен быть совместим с labels (т.е., все измерения должны быть либо 1, или совпадать с соответствующим измерением losses). |
scope | Область действия операций, выполняемых при вычислении потерь. |
loss_collection | Коллекция, в которую будут добавлены потери. |
reduction | Тип сокращения, применяемый к потерям. |
| Возвращаемые значения | |
|---|---|
Числовое значение весовых потерь Tensor. Если reduction равно NONE, это имеет такую же форму, как labels; в противном случае это скаляр. |
| Исключения | |
|---|---|
ValueError | Если формы logits и labels не совпадают или если labels или logits равны None. |
Совместимость с Eager
Аргумент loss_collection игнорируется при выполнении в Eager режиме. Рассмотрите возможность сохранения возвращаемого значения или сбора потерь через tf.keras.Model.
© 2020 The TensorFlow Authors. All rights reserved.
Licensed under the Creative Commons Attribution License 3.0.
Code samples licensed under the Apache 2.0 License.
https://www.tensorflow.org/versions/r2.4/api_docs/python/tf/compat/v1/losses/hinge_loss