Spec-Zone.ru › TensorFlow 1.15

tf.contrib.metrics.auc_using_histogram

AUC, вычисленный с помощью гистограмм.

tf.contrib.metrics.auc_using_histogram(
    boolean_labels, scores, score_range, nbins=100, collections=None,
    check_shape=True, name=None
)

Вместо прямого вычисления AUC, этот оператор сохраняет переменные, содержащие гистограммы оценок, связанных с True и False метками. Сравнивая их, генерируется AUC с некоторой ошибкой дискретизации. См.: "Efficient AUC Learning Curve Calculation" by Bouckaert.

Этот оператор AUC обновляется за O(batch_size + nbins) время и хорошо работает даже при сильном дисбалансе классов. Точность ограничена ошибкой дискретизации из-за конечного числа бинов. Если оценки сконцентрированы в меньшем количестве бинов, точность ниже. Если это вызывает опасения, рекомендуется попробовать различные количества бинов и сравнить результаты.

Аргументы
boolean_labels 1-мерный булевый Tensor. Элемент равен True , если соответствующая запись принадлежит классу.
scores 1-мерный числовой Tensor, имеющий такую же форму, как и boolean_labels.
score_range Tensor формы [2], такого же типа, что и scores. Минимальные и максимальные значения оценки, которые мы ожидаем. Оценки за пределами диапазона будут обрезаны.
nbins Целое число бинов для использования. Точность строго увеличивается по мере увеличения числа бинов.
collections Список ключей коллекций графа. Внутренние переменные гистограмм добавляются в эти коллекции. По умолчанию [GraphKeys.LOCAL_VARIABLES].
check_shape Булево. Если True, выполняется проверка формы оценок и меток во время выполнения.
name Имя этого оператора. По умолчанию "auc_using_histogram".
Возвращаемое значение
auc float32 скаляр Tensor. Получение этого значения преобразует внутренние гистограммы в значение AUC.
update_op Op, при выполнении обновляет внутренние гистограммы.

© 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/r1.15/api_docs/python/tf/contrib/metrics/auc_using_histogram

Spec-Zone.ru

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