/ Модель

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

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

finbert-tone-finetuned-finance-text-classification

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

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

272скачиваний за 30 дней0лайковвеса обновлены3 апр 2023

Бенчмарки

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

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

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

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

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

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

Модель nickmuchi/finbert-tone-finetuned-finance-text-classification разработана как дообученная версия yiyanghkust/finbert-tone для классификации финансовых текстов. Это модель семейства BERT с архитектурой BertForSequenceClassification, содержащая 137 млн параметров и рассчитанная на контекст до 512 токенов.

Модель обучена на объединённом наборе финансовых текстов, включающем financial_phrasebank, FinanceInc/auditor_sentiment и zeroshot/twitter-financial-news-sentiment. Её можно применять для классификации тональности и других меток в финансовых текстах. В карточке приведены результаты оценки: accuracy 0,9097 и F1 0,9102.

Модель принимает текст и выдаёт метки классификации. Карточка указывает английский язык. Доступны веса PyTorch и Safetensors; лицензия в предоставленных данных не указана.

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

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

Ограничения

Ограничение контекста составляет 512 токенов. Результаты оценки на наборе из карточки не гарантируют такое же качество на других финансовых текстах.

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

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

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

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

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

Открыть на Hugging Face

finbert-tone-finetuned-finance-text-classification

This model is a fine-tuned version of yiyanghkust/finbert-tone on the nickmuchi/financial-text-combo-classification dataset which is a combined dataset of financial_phrasebank,FinanceInc/auditor_sentiment and zeroshot/twitter-financial-news-sentiment. It achieves the following results on the evaluation set:

  • Loss: 0.6645
  • Accuracy: 0.9097
  • F1: 0.9102
  • Precision: 0.9110
  • Recall: 0.9097

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: 128
  • eval_batch_size: 128
  • seed: 42
  • optimizer: Adam with betas=(0.9,0.999) and epsilon=1e-08
  • lr_scheduler_type: linear
  • num_epochs: 20
  • mixed_precision_training: Native AMP

Training results

Training LossEpochStepValidation LossAccuracyF1PrecisionRecall
No log1.01410.39340.84310.84270.84560.8431
No log2.02820.32140.88430.88430.88670.8843
No log3.04230.33020.88820.89020.89650.8882
0.44444.05640.36110.89800.89930.90260.8980
0.44445.07050.40060.89750.89870.90140.8975
0.44446.08460.45170.90370.90430.90570.9037
0.44447.09870.53240.90270.90350.90570.9027
0.04068.011280.53080.90630.90740.90980.9063
0.04069.012690.55860.90810.90840.90890.9081
0.040610.014100.57830.90760.90800.90860.9076
0.012111.015510.57410.91150.91160.91210.9115
0.012112.016920.62880.91040.91080.91150.9104
0.012113.018330.63280.90500.90590.90780.9050
0.012114.019740.68870.90420.90540.90880.9042
0.006315.021150.63450.90860.90940.91090.9086
0.006316.022560.65450.91020.91030.91080.9102
0.006317.023970.65850.90860.90920.91030.9086
0.003318.025380.66760.90810.90870.90980.9081
0.003319.026790.66140.91100.91130.91190.9110
0.003320.028200.66450.90970.91020.91100.9097

Framework versions

  • Transformers 4.25.1
  • Pytorch 1.13.1+cu116
  • Datasets 2.8.0
  • Tokenizers 0.13.2

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей