Multi class emotion classification for text in russian

1

Как использовать BERT Classifier для multi class классификаций текста? У меня есть свой датасет, нужно тренировать модель на этом датасете.
Пример Input:
Я сегодня чувствую себя не очень хорошо
Output:
Sadness

Классов должно быть 5 или 6

Знаю что есть rusentiment_bert.json. Это как я понимаю pretrained и здесь только Positive neutral negative speech skip, а мне надо чтобы были эмоций типа (радость, печаль итп)

Мне получается нужно быть изменить конфиг rusentiment_bert.json? Если да – то как?
Прошу помочь c гайденсом как работает весь процесс.

2

Добрый день!

Вам нужно заменить пути к датасету, названия столбцов, разделители, если необходимо.
Также следует заменить пути к моделям, чтобы автоматически не подгружать веса rusentiment модели.

3

Спасибо за быстрый ответ.
Нашел данный туториал dp_tutorials/Tutorial_3_RU_Fine_tuning_BERT_classifier.ipynb at master · deepmipt/dp_tutorials · GitHub

Подскажите пожалуйста, является ли данный туториал ответом на мой вопрос? Естественно с небольшими изменениями