/ Модель

Характеристики

Контекст
512 токенов
Параметры
130 млн
Лицензия
CC BY 4.0
Архитектура
bert · dense
Опубликована
28 мая 2024

bert-suicide-detection-hk

Классификация текстаТекст → label

Дообученная BERT-модель для классификации текста с целью выявления суицидальной тематики.

9скачиваний за 30 дней0лайковвеса обновлены6 июн 2024

Бенчмарки

Как модель справилась с задачами пользователей ailist и что о ней публикуют открытые лидерборды

Публичные оценки

Из открытых лидербордов. Сравнивайте результаты одного теста при одинаковых условиях — отсутствие оценки не значит, что модель хуже.

В подключённых источниках результатов для этой модели пока нет.

Что умеет модель

Описание составлено по карточке модели и метаданным Hugging Face

wcyat/bert-suicide-detection-hk — модель классификации текста на базе BERT, дообученная разработчиком wcyat для выявления суицидальной тематики в тексте. Она предназначена для автоматической маркировки текстов по категориям, связанным с риском суицида; может использоваться как вспомогательный фильтр для последующей проверки специалистом. Модель содержит 129,9 млн параметров, принимает текстовые последовательности длиной до 512 токенов и выдаёт результат классификации. Репозиторий использует формат safetensors.

Модель опубликована по лицензии CC BY 4.0. Карточка не описывает обучающие данные, классы классификации, метрики качества или ограничения применения, поэтому надёжность результатов и пригодность для оценки риска конкретного человека по имеющимся сведениям определить нельзя.

Сценарии применения

  • выявление текстов с упоминанием суицида
  • маркировка сообщений для последующей проверки специалистом
  • первичная фильтрация публикаций на суицидальную тематику

Ограничения

В карточке нет описания обучающих данных, метрик, классов и проверенных ограничений; качество выявления риска по этим сведениям оценить нельзя. Результат классификации не следует считать диагнозом или самостоятельной оценкой риска человека.

Карточка модели

README от автора на Hugging Face, как он его опубликовал

Открыть на Hugging Face

bert-suicide-detection-hk

This model is a fine-tuned version of wcyat/bert-suicide-detection-hk on the None dataset.

Model description

More information needed

Intended uses & limitations

More information needed

Training and evaluation data

More information needed

Training procedure

Training hyperparameters

The following hyperparameters were used during training:

  • learning_rate: 2e-05
  • train_batch_size: 4
  • eval_batch_size: 4
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 5

Framework versions

  • Transformers 4.41.2
  • Pytorch 2.3.0+cu121
  • Datasets 2.19.2
  • Tokenizers 0.19.1

Как запустить

На своей GPU в ailist. Цена — по текущим предложениям провайдеров

GPU в ailist

Универсальный рантайм · text-classification

Можно запустить

Подбираем GPU…

Класс архитектуры
BertForSequenceClassification
Ревизия весов
242f491

0 моделей