В проекте изучается возможность предварительного обучения графовых нейронных сетей на исходном коде. Проект состоит из нескольких частей:

• Инструмента для извлечения графовых представлений из исходного кода на различных языках.
• Реализации графовой нейронной сети для различных задач ML4SE и целей предварительного обучения. Мы реализовали и оценили 8 типов графовых нейронных сетей на основе библиотеки PyTorch-Geometric с учетом их масштабирования.
• Построение фреймворка / конфигурируемого пайпайна для удобства экспериментов с задачами ML4SE. Фреймворком уже можно пользоваться.
• Предложения по улучшению архитектуры архитекруты графовых нейронных сетей и задач обучения.