Как извлечь TP, FN, FP из классификационного отчета - PullRequest
0 голосов
/ 10 апреля 2020

Я знаю, что могу вывести содержимое sklearn classification_report в виде dict и извлечь микро / макро усредненные оценки вместе с оценками для отдельных ярлыков.

Как я могу получить значения для соответствующих входных данных матрицы смешения, в частности, TP, FN и FP, которые go, в вычислениях для макро-усредненных мер? У меня есть векторы для y_pred и y_true, поэтому я предполагаю, что есть способ получить желаемые параметры, которые я хочу.

Это для проблемы двоичной классификации, но она мне также понадобится для мультиклассовая задача, а также.

1 Ответ

0 голосов
/ 10 апреля 2020

Понял: tn, fp, fn, tp = confusion_matrix([0, 1, 0, 1], [1, 1, 1, 0]).ravel()

и sklearn.metrics.multilabel_confusion_matrix

...