/ Модель

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

Контекст
514 токенов
Параметры
152 млн
Лицензия
MIT
Архитектура
roberta · dense
Опубликована
15 июн 2022

roberta-base-qqp

Классификация текстаТекст → label6 профессий

Дообученная на GLUE QQP модель RoBERTa определяет, являются ли два английских вопроса дубликатами.

179скачиваний за 30 дней0лайковвеса обновлены6 дек 2022

Бенчмарки

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

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

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

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

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

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

JeremiahZ/roberta-base-qqp — модель классификации текстовых пар, созданная дообучением RoBERTa-base на наборе GLUE QQP. Она предназначена для определения, задают ли два английских вопроса один и тот же вопрос, например, при поиске дубликатов в FAQ или объединении похожих обращений. Модель построена на архитектуре RoBERTa для классификации последовательностей, содержит 151,8 млн параметров и принимает текстовые входные данные с контекстом до 514 токенов.

В карточке указаны результаты оценки на тестовом наборе: accuracy 0,9153 и F1 0,8867. Модель поддерживает английский язык и опубликована в формате PyTorch. Лицензия — MIT. Карточка не описывает дополнительные сценарии использования и ограничения; качество на других данных может отличаться от результатов на GLUE QQP.

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

  • поиск дублирующихся вопросов в FAQ
  • сопоставление похожих обращений в службу поддержки
  • объединение повторяющихся вопросов в базе знаний

Ограничения

Модель обучена на английском наборе GLUE QQP, поэтому её качество на других языках и типах текстовых пар не подтверждено. В карточке не приведены дополнительные сведения об ограничениях.

Для каких профессий подходит6

Разделы каталога профессий, в задачах которых используется эта модель

Все профессии

Также подходит для

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

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

Открыть на Hugging Face

roberta-base-qqp

This model is a fine-tuned version of roberta-base on the GLUE QQP dataset. It achieves the following results on the evaluation set:

  • Loss: 0.4435
  • Accuracy: 0.9153
  • F1: 0.8867
  • Combined Score: 0.9010

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: 16
  • eval_batch_size: 8
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • lr_scheduler_warmup_ratio: 0.06
  • num_epochs: 10.0

Training results

Training LossEpochStepValidation LossAccuracyF1Combined Score
0.27511.0227410.30570.89050.85120.8709
0.24432.0454820.25300.90050.87100.8857
0.21573.0682230.26430.90700.87690.8919
0.18384.0909640.28060.91090.88150.8962
0.1465.01137050.32770.91130.88090.8961
0.12626.01364460.39390.91130.88120.8962
0.08677.01591870.44350.91530.88670.9010
0.07578.01819280.48120.91470.88440.8996
0.04799.02046690.50810.91510.88710.9011
0.037910.02274100.56470.91490.88580.9003

Framework versions

  • Transformers 4.20.0.dev0
  • Pytorch 1.11.0+cu113
  • Datasets 2.1.0
  • Tokenizers 0.12.1

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

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

GPU в ailist

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

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

Подбираем GPU…

Класс архитектуры
RobertaForSequenceClassification
Ревизия весов
749a735

0 моделей