Участники проекта:
- Жучков Е.А. ИДБ-18-09
- Жучкова Т.Г. ИДБ-18-09
Приложение получает от пользователя метод или функцию на одном из поддерживаемых языков программирования. Обученная модель генерирует комментарий на основе данного кода, описывающий его функционал.
Поскольку данная проблема сводится к задаче машинного перевода, была выбрана seq2seq архитектура на основе трансформера BART. Для обучения были взяты данные из датасета CodeSearchNet. На данный момент для генерации доступны только функции на языке Python. Приложение использует Flask в качестве веб-фреймворка.
- Клонировать проект на локальный компьютер.
git clone https://github.com/EgorZhuchkov/code_to_docstring
cd code_to_docstring
- Убедиться, что на компьютере установлены необходимые зависимости, указанные в requirements.txt.
pip install -r requirements.txt
- Запустить файл app.py в корне проекта.
python app.py
- Перейти по адресу указанному в консоли. Например:
Running on http:https://localhost:5000/
Предупреждение:
При запуске приложение загружает из облака модель и чекпоинт. Это может занять некоторое время.