ValueError: Найдены входные переменные с непоследовательным количеством выборок: [6, 1]

Question

Я пробиваю кнопку классификации изображений. Я хотел бы рассчитать точность и вспомнить для каждого изображения. В настоящее время для каждого изображения у меня есть два .txt: один указывает GT и один указывает обнаружение. Пример:

GT: [59, 59, 59]
DETECTION: [30, 30, 1]

Как рассчитать метрики, если длина GT меньше или больше длины обнаружения?

GT: [64, 64, 64, 64, 64]
DETECTION: [34]

или

GT: [59, 59, 59]
PRED: [30, 30, 1, 59]

Я пытался использовать эту функцию: precision = precision_score(GT, DETECTION, average = 'macro'), но когда len(GT) отличается от len(DETECTION), он не может рассчитать метрики ...

ValueError: Found input variables with inconsistent numbers of samples: [6, 1]

Кто-нибудь знает, как мне помочь? Заранее спасибо

Answer 1

Нет, вы должны использовать массив одинаковой формы в обоих случаях. Например, если у вас есть K классов, тогда оба вектора должны иметь форму (1, K). Тогда значения будут 0 или 1, в зависимости от того, присутствует класс или нет. Вы должны кодировать категории.