/ Модель

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

Контекст
512 токенов
Параметры
137 млн
Архитектура
bert · dense
Опубликована
2 мар 2022

autonlp-bank-transaction-classification-5521155

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

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

54скачиваний за 30 дней26лайковвеса обновлены17 фев 2024

Бенчмарки

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

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

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

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

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

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

mgrella/autonlp-bank-transaction-classification-5521155 — модель классификации банковских транзакций, обученная с помощью AutoNLP. Она решает задачу многоклассовой классификации текста и предназначена для присвоения транзакционным записям категорий по их текстовому описанию. Модель построена на архитектуре BERT для классификации последовательностей, содержит 137,1 млн параметров и принимает текст длиной до 512 токенов.

В карточке указан итальянский язык. На валидации точность составила 0,822, а macro F1 — 0,571; разница между этими показателями указывает, что качество может заметно различаться между классами. Модель опубликована в формате PyTorch и Safetensors. Лицензия в предоставленных данных не указана.

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

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

Ограничения

Macro F1 на валидации составляет 0,571, что указывает на существенно менее равномерное качество по классам, чем общая точность 0,822. Максимальная длина входа — 512 токенов; в предоставленных данных лицензия не указана.

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

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

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

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

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

Открыть на Hugging Face

Model Trained Using AutoNLP

  • Problem type: Multi-class Classification
  • Model ID: 5521155

Validation Metrics

  • Loss: 1.3173143863677979
  • Accuracy: 0.8220706757594545
  • Macro F1: 0.5713688384455807
  • Micro F1: 0.8220706757594544
  • Weighted F1: 0.8217158913702755
  • Macro Precision: 0.6064387992817253
  • Micro Precision: 0.8220706757594545
  • Weighted Precision: 0.8491515834140735
  • Macro Recall: 0.5873349311175117
  • Micro Recall: 0.8220706757594545
  • Weighted Recall: 0.8220706757594545

Usage

You can use cURL to access this model:

$ curl -X POST -H "Authorization: Bearer YOUR_API_KEY" -H "Content-Type: application/json" -d '{"inputs": "I love AutoNLP"}' https://api-inference.huggingface.co/models/mgrella/autonlp-bank-transaction-classification-5521155

Or Python API:

from transformers import AutoModelForSequenceClassification, AutoTokenizer

model = AutoModelForSequenceClassification.from_pretrained("mgrella/autonlp-bank-transaction-classification-5521155", use_auth_token=True)

tokenizer = AutoTokenizer.from_pretrained("mgrella/autonlp-bank-transaction-classification-5521155", use_auth_token=True)

inputs = tokenizer("I love AutoNLP", return_tensors="pt")

outputs = model(**inputs)

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей