Использую этот конфиг для обнаружения оскорблений insults_kaggle_conv_bert , модель выдает ответ в виде оскорбление или не оскорбление. Подскажите, откуда из кода вытащить степень уверенности в оскорблении ? чтобы результат примерно выглядел как “Insult, value - 85%”
Чтобы добавить вывод вероятности, добавьте в out пайплайна insults_kaggle_conv_bert параметр "y_pred_probas", т.е. 102 строка insults_kaggle_conv_bert.json должна быть заменена на "y_pred_labels", "y_pred_probas". Тогда модель будет возвращать не только название класса, но и массив вероятностей.