Библиотека Py-Boost оптимизирует проведение градиентного бустинга — одного из алгоритмов машинного обучения на табличных данных. В ней используется алгоритм SketchBoost, который позволяет значительно ускорить обучение моделей.

Py-Boost работает только с использованием графических ускорителей и использует написанные на языке Python библиотеки для работы с ними, такие как СuPy и Numba.

Библиотека была представлена Лабораторией искусственного интеллекта «Сбера» в июне 2023 года. Доступна как открытое ПО.

Репозиторий на GitHub

Статья в сборнике материалов конференции NeurIPS

👤 Антон Вахрушев

• Смотреть

#искусственный_интеллект#нейросети#машинное_обучение#open_source

#Сбербанк