/ChatClassificationTask

Test task for chat text classification problem

Primary LanguageJupyter NotebookCreative Commons Zero v1.0 UniversalCC0-1.0

ChatClassificationTask

Test task for chat text classification problem

Обучение моделей и эксперименты производятся в Google Colab

Google Colab

Open TextClassification.ipynb in Google Colab!

Open In Colab

API

Интерфейс модели реализован через FastAPI. Для запуска микросервиса требуется docker и docker-compose.

  1. Чтобы развернуть API локально на компьютере клонируйте код локально на компьютер

git clone https://github.com/mishagrol/ChatClassificationTask.git

  1. Перейди в директорию с кодом

cd ChatClassificationTask

  1. Запустите docker-compose, который соберет image и запустит сервис на uvicorn

docker-compose up

  1. В случае удачного запуска вы увидете локальный адресс на котором работает код

~> Uvicorn running on http://0.0.0.0:8000

  1. Чтобы протестировать работу API перейдите по ссылке Docs

  2. Вам станет доступна панель Swagger в котором вы можете передать сообщение из чата в метод POST - /predict_message_type

Пример документации

Awesome