Реализовано

Простейший телеграм чат-бот, который ведёт диалог от лица компании MusicRocket, он умеет:

1. Определить какой услугой хочет воспользоваться пользователь, и сообщит ему цену
   Сейчас у чат-бота строгое ограничение на определение желания пользователя, это решается замены классификатора на Multi-label, для этого нужен датасет реальных писем/сообщений пользователей.
2. Поддержать простой диалог

Опробовать бота можно тут

Возможные следующие шаги

1. Применить Self-Supervised Learning, например некоторые функция шума из статьи про Bart на датасете, для повышения устойчивости модели
2. Заменить затычку в виде rugpt, на обученную Data-to-Text Generation допустим скармливать ему контекст и данные об услугах выбранных пользователем, чтобы получался более живой ответ
3. Сделать графовую систему сценариев, тогда можно будет уточнять недостающую информацию, например имя человека, название группы, удобный вариант связи, в какое время ему будет удобно посетить студию, попросить прислать текст и получить фидбек.

Usage

установка зависимостей

pip install -r requirements.txt

Для получения весов нужно запустить train_classification.py, после можно запускать бота telegram_bot.py