Центр Big Data МТС открыл исходный код для создания графовых нейросетей

Библиотека CoolGraph предназначена как для экспертов, так и для начинающих специалистов в области глубокого обучения нейросетей, имеющих доступ к данным графов, с помощью которых они могут строить различные модели. Специалисты Big Data МТС собрали в библиотеке свои лучшие наработки в области построения кастомных моделей.

Простота использования CoolGraph заключается в том, что библиотека позволяет легко задать архитектуру сети и автоматически подобрать гипер-параметры. Также разработка содержит примеры запуска и использования моделей различной конфигурации. Использовать её можно с помощью двух строк кода, а отслеживать все результаты экспериментов пользователь может в MLFlow.

«Для достижения прорывов в области машинного обучения необходимо большое количество талантливых людей, экспериментирующих с технологиями. Наша библиотека решает эту задачу за счет снижения порога входа в графовые нейросети. Эта технология, с одной стороны, объединяет в себе известные модели глубокого обучения, а с другой стороны, дает инструмент для новых исследований» — комментирует директор центра Big Data МТС Виктор Кантор.

В дальнейшем центр Big Data МТС планирует пополнять библиотеку CoolGraph и открывать доступ к другим своим разработкам.