Узнайте, как матрица путаницы оценивает эффективность классификации. Изучите TP, FP, TN и FN, чтобы оптимизировать модели Ultralytics для повышения точности.
Матрица путаницы — это инструмент измерения эффективности для задач классификации в машинном обучении, где результатом может быть два или более классов. Это таблица с четырьмя различными комбинациями прогнозируемых и фактических значений, служащая основой для визуализации данных при оценке модели. В отличие от простой точности, которая может вводить в заблуждение, если набор данных несбалансирован, матрица путаницы предоставляет подробную разбивку того, где модель компьютерного зрения (CV) допускает ошибки. Сравнивая прогнозы с метки «истинного значения», разработчики могут определить, путает ли система два конкретных класса или не может detect полностью.
Сама матрица обычно делится на четыре квадранта для бинарной классификации, хотя она расширяется для многоклассовых задач, таких как те, которые обрабатываются Ultralytics . Эти четыре компонента представляют собой пересечение того, что предсказала модель, с тем, что действительно существует на изображении.
Исходные числа в матрице путаницы используются для расчета более сложных метрик, которые описывают производительность модели. Понимание этих производных имеет важное значение для оптимизации нейронных сетей.
Конкретная стоимость ошибок, определяемая матрицей путаницы, диктует, как модели настраиваются для разных отраслей.
В области ИИ в здравоохраненииматрица путаницы является вопросом безопасности. При обучении модели для анализа медицинских изображений с целью detect , ложное отрицательное (пропуск опухоли) гораздо хуже, чем ложное положительное (пометка доброкачественного пятна для осмотра врачом). Поэтому инженеры отдают приоритет воспроизводимости над точностью в этих матрицах, чтобы гарантировать, что никакие потенциальные риски для здоровья не будут упущены.
И наоборот, в контроле качества производства, ключевую роль играет эффективность. Если система, классифицирующая детали на конвейере, генерирует слишком много ложных срабатываний (помечая исправные детали как дефектные), это приводит к ненужным потерям и замедлению производства. В этом случае матрица путаницы помогает инженерам настроить модель для максимальной точности, гарантируя, что отбракованные детали действительно являются дефектными, и оптимизируя автоматизированныерабочие процессы машинного обучения .
При использовании современных фреймворков генерация этой матрицы часто является частью стандартного конвейера валидации. Пример
ниже демонстрирует, как проверить
Модель YOLO26 и получить доступ к данным матрицы путаницы
с помощью ultralytics пакет.
from ultralytics import YOLO
# Load a pre-trained YOLO26 model
model = YOLO("yolo26n.pt")
# Validate the model on the COCO8 dataset
# This automatically generates and plots the confusion matrix
metrics = model.val(data="coco8.yaml")
# Access the confusion matrix object directly
print(metrics.confusion_matrix.matrix)
Важно отличать матрицу путаницы от схожих терминов оценки.