#multilabel_classification — Public Fediverse posts

Почему «чем проще, тем лучше» не работает на ИИ-классификаторе

Обучил multi-label классификатор на 15 классов для модерации Discord-сообщества, получил micro F1 = 0.9358 — цифра, с которой можно закрывать задачу и не разбираться дальше. Но стоило посмотреть на precision и recall по каждому классу отдельно, как выяснилось: recall на TOXIC — около 0.78, а для части редких меток test split вообще не подтверждает качество — положительных примеров там почти нет. Разбираю на реальных цифрах и коде: почему агрегированная метрика такое скрывает, как считать вес классов через pos_weight при сильном дисбалансе, почему checkpoint стоит выбирать по macro F1, а не по training loss, и где принцип «чем проще — тем лучше» перестаёт работать при оценке качества классификатора. Подробнее

https://habr.com/ru/articles/1059444/

#machine_learning #multilabel_classification #ruBERT #class_imbalance #nlp #pytorch #bert #text_classification