/ Модель

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

Контекст
514 токенов
Параметры
305 млн
Архитектура
xlm-roberta · dense
Опубликована
12 мая 2022

lang-classifier-roberta

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

Дообученная XLM-RoBERTa определяет язык текста среди 11 языков, включая русский, украинский и другие европейские и региональные языки.

16скачиваний за 30 дней5лайковвеса обновлены7 янв 2023

Бенчмарки

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

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

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

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

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

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

nikitast/lang-classifier-roberta — модель классификации текста по языку, созданная дообучением RoBERTa. Она принимает текст и определяет, к одному из поддерживаемых языков он относится: русскому, украинскому, белорусскому, казахскому, азербайджанскому, венгерскому, армянскому, грузинскому, ивриту, английскому или немецкому. Модель основана на XLM-RoBERTa, содержит около 305,2 млн параметров и рассчитана на последовательности длиной до 514 токенов.

Модель дообучена на небольших выборках Open Subtitles, OSCAR и Tatoeba — примерно по 9 тысяч примеров на язык. В README приведены результаты проверки на отдельных выборках примерно по тысяче примеров на язык; для показанных классов F1 составляет от 0,976 до 0,998. Размер обучающих выборок ограничивает уверенность в качестве на текстах, которые отличаются от использованных данных. Репозиторий содержит веса PyTorch; сведения о лицензии в доступных данных отсутствуют.

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

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

Ограничения

Обучение проводилось на небольших выборках из Open Subtitles, OSCAR и Tatoeba, поэтому качество на других стилях и типах текста может отличаться. Указанное окно контекста ограничено 514 токенами.

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

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

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

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

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

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

Открыть на Hugging Face

RoBERTa for Single Language Classification

Training

RoBERTa fine-tuned on small parts of Open Subtitles, Oscar and Tatoeba datasets (~9k samples per language).

data sourcelanguage
open_subtitleska, he, en, de
oscarbe, kk, az, hu
tatoebaru, uk

Validation

The metrics obtained from validation on the another part of dataset (~1k samples per language).

indexclassf1-scoreprecisionrecallsupport
0az0.9980.9971.0997
1be0.9960.9980.9941004
2de0.9760.9660.987979
3en0.9760.9860.9671020
4he1.01.00.9991001
5hy0.9940.9910.998993
6ka0.9990.9990.9991000
7kk0.9960.9980.9931005
8uk0.9820.9970.9681030
9ru0.9820.9680.997971
10macro_avg0.990.990.9910000
11weighted avg0.990.990.9910000

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

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

GPU в ailist

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

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

Подбираем GPU…

Класс архитектуры
XLMRobertaForSequenceClassification
Ревизия весов
7c97c98

0 моделей