Показатели sklearn f1_score: pos_label valueError

Question

Для приведенной ниже строки кода для получения метрик я получаю упомянутое значение ValueError

f1_test.append (f1_score (y_test, y_pred_test, pos_label = "str"))

ValueError: pos_label = 'str' не является допустимой меткой: массив (['<= 50K', '> 50K'], dtype = '

Мои данные имеют значение Income vlaue <= 50k или> 50k.

Пожалуйста, сообщите.

Answer 1

Параметр pos_label используется для указания положительного класса. В ваших данных ваши метки основаны на строках '<= 50K' и '> 50K'.

Вы можете выполнить одно из следующих действий:

1. Оставьте pos_label значение по умолчанию, установите pos_label = 1.
2. Установите pos_label = '> 50K' или '<= 50K', в зависимости от вашего класса + ve. Позитивный класс обычно относится к лучшему результату, например, доход, который вы пытаетесь предсказать, лучше, если его больше, чем 50K или меньше, чем <= 50K? </li>