БЕРТ Уверенность (iOS)

Question

В настоящее время я интегрирую модель BERT, указанную в https://developer.apple.com/machine-learning/models/#text, в приложение iOS, и мне было трудно удалить ответы с низкой степенью достоверности.

Я использовал пример кода найденный по ссылке выше, но так как я хотел ответить на вопросы, основанные на больших объемах текста, я набрал oop по массиву параграфов и предсказать ответ для каждого из них. Однако модель не возвращает nil или «No Answer», если ответ не найден, и вместо этого возвращает (казалось бы) случайную подстроку. Я предполагаю, что я пытаюсь спросить: возможно ли получить доступ к определенности ответа BERT, чтобы отфильтровать маловероятные результаты? Или есть другой способ заставить BERT возвращать результаты только выше установленного порога достоверности?

Answer 1

После нескольких часов поиска я нашел решение. По иронии судьбы это заняло всего три строки кода, но в любом случае:

if bestSum < 7.5 {
    return nil
}

Я реализовал это в методе findBestLogitPair() в файле BERTOutput.swift, как это предусмотрено в примере кода Apple для анализа текста с использованием BERT. , Теперь я обнаружил, что слово lo git означает некоторую вероятность в статистике, но, будучи программистом, я понятия не имел!