Обнаружение исправлений

Добрый день

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

Добрый день, @Igor

Чтобы добавить вывод вероятности, добавьте в out пайплайна insults_kaggle_conv_bert параметр "y_pred_probas", т.е. 102 строка insults_kaggle_conv_bert.json должна быть заменена на "y_pred_labels", "y_pred_probas". Тогда модель будет возвращать не только название класса, но и массив вероятностей.

Спасибо, получилось.
[
“Insult”,
[
0.028377605602145195,
0.9716224074363708
]

Кстати, а с какой цифры у вас считается оскорбление