Spec-Zone.ru › TensorFlow 2.3

tf.compat.v1.losses.hinge_loss

Добавляет потерю Хинге к процедуре обучения.

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} при вычислении потери Хинге.
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.3/api_docs/python/tf/compat/v1/losses/hinge_loss

Spec-Zone.ru

Настройки Оффлайн Что нового Помощь О нас
Spec-Zone .ru
спецификации, руководства, описания, API