Вероятностные выходные оценки модели классификации текста имеют максимальные значения - PullRequest
0 голосов
/ 28 мая 2020

Я обучаю модель классификации текста, используя предварительно обученную модель BERT из библиотеки huggingface. Он работает очень хорошо, и я получаю точность проверки 90% или выше. Но когда я смотрю на оценки выходных вероятностей для каждой категории, кажется, что они имеют максимальное значение. Для моих четырех категорий обычный пример выглядит так: [0,43, 0,19, 0,19, 0,19]. Если вероятность категории с наивысшей вероятностью ни в коем случае не превышает 0,43. Это остается, когда я тренируюсь для новых эпох. Мои вопросы задают чисто из любопытства, поскольку модель, которая у меня есть, может быть развернута. Кто-нибудь знает, является ли такое поведение нормальным или это проблема, которую необходимо решить?

...