tf.contrib.metrics.cohen_kappa
Вычисляет коэфффициент согласия Коэна.
tf.contrib.metrics.cohen_kappa(
labels, predictions_idx, num_classes, weights=None, metrics_collections=None,
updates_collections=None, name=None
)
Коэффициент согласия Коэна — это статистическая величина, которая измеряет согласованность между оценщиками.
Функция cohen_kappa вычисляет матрицу ошибок и создаёт три локальные переменные для вычисления коэффициента согласия Коэна: po, pe_row, и pe_col, которые соответствуют диагональной части, суммам строк и столбцов матрицы ошибок, соответственно. Эта величина в конечном итоге возвращается как kappa, идемпотентная операция, которая вычисляется следующим образом:
pe = (pe_row * pe_col) / N k = (sum(po) - sum(pe)) / (N - sum(pe))
Для оценки метрики по потоку данных функция создаёт операцию update_op, которая обновляет эти переменные и возвращает kappa. update_op весит каждое предсказание соответствующим значением в weights.
Ожидается, что метки классов начинаются с 0. Например, если num_classes было равно трём, то возможные метки будут [0, 1, 2].
Если weights равно None, веса по умолчанию равны 1. Используйте веса 0, чтобы замаскировать значения.
Примечание: Эквивалентно sklearn.metrics.cohen_kappa_score, но метод пока не поддерживает взвешенную матрицу.
| Аргументы | |
|---|---|
labels | 1-мерный Tensor из вещественных меток для задачи классификации. Должен быть одного из следующих типов: int16, int32, int64. |
predictions_idx | 1-мерный Tensor из индексов предсказанных классов для данной задачи классификации. Должен иметь тот же тип, что и labels. |
num_classes | Возможное количество меток. |
weights | Необязательные Tensor с формой, соответствующей predictions. |
metrics_collections | Необязательный список коллекций, к которым kappa должны быть добавлены. |
updates_collections | Необязательный список коллекций, к которым update_op должны быть добавлены. |
name | Необязательное имя области имен переменных. |
| Возвращаемые значения | |
|---|---|
kappa | Скалярное значение с плавающей точкой Tensor, представляющее текущий коэффициент согласия Коэна. |
update_op | Operation который увеличивает переменные po, pe_row и pe_col соответствующим образом и значение которого совпадает с kappa. |
| Исключения | |
|---|---|
ValueError | Если num_classes меньше 2, или predictions и labels имеют несовпадающие формы, или если weights не None и его форма не совпадает с predictions, или если либо metrics_collections или updates_collections не являются списком или кортежем. |
RuntimeError | Если включено выполнение в режиме eager. |
© 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/cohen_kappa