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

Контекст: 512 токенов
Параметры: 137 млн
Архитектура: bert · dense

Ссылки

Hugging Face

industry-classification-api

sampathkethineedi

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

BERT-модель классифицирует описания компаний по 62 отраслевым тегам на основе примеров индийского бизнеса.

13скачиваний за 30 дней16лайковвеса обновлены19 мая 2021

Бенчмарки

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

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

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

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

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

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

Модель sampathkethineedi/industry-classification-api — BERT-классификатор, разработанный для отнесения описаний компаний к одному из 62 отраслевых тегов. Он обучен на 7000 примерах описаний индийского бизнеса с соответствующими метками и предназначен для классификации бизнес-профилей по отраслям.

Модель содержит 136,7 млн параметров, принимает текст и поддерживает контекст до 512 токенов. Доступны веса PyTorch и Flax; карточка указывает английский язык. Модель подходит для автоматической категоризации компаний, каталогов организаций и бизнес-справочников. Обучающая выборка охватывает только индийские компании, поэтому качество на компаниях и описаниях из других стран может быть ограничено. Лицензия в предоставленных данных не указана.

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

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

Ограничения

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

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

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

Открыть на Hugging Face

industry-classification-api

Model description

BERT Model to classify a business description into one of 62 industry tags. Trained on 7000 samples of Business Descriptions and associated labels of companies in India.

How to use

PyTorch only

from transformers import AutoTokenizer, AutoModelForSequenceClassification, pipeline

tokenizer = AutoTokenizer.from_pretrained("sampathkethineedi/industry-classification")  
model = AutoModelForSequenceClassification.from_pretrained("industry-classification")

industry_tags = pipeline('sentiment-analysis', model=model, tokenizer=tokenizer)
industry_tags("Stellar Capital Services Limited is an India-based non-banking financial company ... loan against property, management consultancy, personal loans and unsecured loans.")

'''Ouput'''
[{'label': 'Consumer Finance', 'score': 0.9841355681419373}]

Limitations and bias

Training data is only for Indian companies

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

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

GPU в ailist

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

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

Подбираем GPU…

Класс архитектуры: BertForSequenceClassification
Ревизия весов: 126bd5e