/ Модель

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

Контекст
512 токенов
Параметры
137 млн
Лицензия
Apache 2.0
Архитектура
bert · dense
Базовая модель
bert-base-uncased
Опубликована
18 мар 2024

bert-petco-emailbody-ctr

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

Дообученная модель BERT для классификации текстов, связанная по названию с телами писем Petco.

13скачиваний за 30 дней0лайковвеса обновлены19 мар 2024

Бенчмарки

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

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

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

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

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

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

yimiwang/bert-petco-emailbody-ctr — дообученная модель на архитектуре BERT для классификации текста. В метаданных указана задача text-classification, а размер модели составляет 136,7 млн параметров; базовая модель — bert-base-uncased. Модель принимает текст и выдаёт результат классификации. Контекст — до 512 токенов.

В README приведены метрики на оценочном наборе: MSE 0,0143, RMSE 0,1197, MAE 0,0727 и R² 0,7276. Название указывает на применение к телам писем Petco, однако описание назначения и данных обучения отсутствует, поэтому нельзя подтвердить конкретные классы или целевую переменную. Распространяется по лицензии Apache-2.0.

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

  • классификация текста тела письма
  • оценка целевого показателя по тексту письма
  • эксперименты с классификацией email-текстов

Ограничения

README не описывает конкретную целевую переменную, метки, датасет и сценарии применения; приведённые метрики сами по себе не позволяют установить семантику предсказания. Контекст ограничен 512 токенами.

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

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

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

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

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

Открыть на Hugging Face

bert-petco-emailbody-ctr

This model is a fine-tuned version of bert-base-uncased on the None dataset. It achieves the following results on the evaluation set:

  • Loss: 0.0143
  • Mse: 0.0143
  • Rmse: 0.1197
  • Mae: 0.0727
  • R2: 0.7276

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

Training results

Training LossEpochStepValidation LossMseRmseMaeR2
0.11871.0160.04740.04740.21760.09060.0990
0.10422.0320.04470.04470.21150.14150.1490
0.06413.0480.03150.03150.17740.09660.4012
0.05624.0640.02780.02780.16670.09020.4716
0.04985.0800.02780.02780.16690.09140.4702
0.03186.0960.02290.02290.15120.09910.5652
0.02397.01120.02750.02750.16580.10250.4770
0.01178.01280.05130.05130.22640.09460.0248
0.01379.01440.03710.03710.19260.08670.2940
0.012510.01600.02870.02870.16940.07690.4538
0.007711.01760.03320.03320.18210.08030.3691
0.004912.01920.02250.02250.15010.09700.5715
0.007413.02080.01850.01850.13600.08220.6482
0.004614.02240.02140.02140.14640.07340.5923
0.00515.02400.01520.01520.12340.07300.7104
0.005916.02560.01430.01430.11970.07270.7276
0.00517.02720.02490.02490.15770.07440.5271
0.004218.02880.02670.02670.16350.07620.4911
0.003919.03040.02830.02830.16830.07760.4609
0.004520.03200.02740.02740.16540.07560.4795

Framework versions

  • Transformers 4.38.2
  • Pytorch 2.2.1+cu121
  • Datasets 2.18.0
  • Tokenizers 0.15.2

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

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

GPU в ailist

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

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

Подбираем GPU…

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

0 моделей