Suicidal-BERT — англоязычная модель классификации текста, разработанная gohjiayi на основе google-bert/bert-base-uncased и дообученная для определения суицидальной тематики в последовательности слов. Она относит текст к одному из двух классов: суицидальный или несуицидальный. Модель предназначена для анализа англоязычных публикаций и текстов, в том числе сообщений в социальных сетях.

Модель построена на архитектуре BERT для классификации последовательностей, содержит 136,7 млн параметров и принимает текст с контекстом до 512 токенов. В README приведены результаты оценки на использованном наборе данных: accuracy 0,9757, recall 0,9669, precision 0,9701 и F1 0,9685. Данные собраны с Reddit и поровну распределены между двумя классами; дообучение проводилось в течение одной эпохи. Модель распространяется по лицензии MIT.

Показатели из README относятся к указанному набору данных и не подтверждают качество на других источниках или в реальных сценариях. Классификацию не следует использовать как клиническую оценку или единственное основание для решений, связанных с безопасностью; возможны ошибки и смещение из-за состава обучающих данных.