PyTorch Geometric Temporal — проект, расширяющий возможности PyTorch Geometric для работы с динамическими графами

PyTorch Geometric Temporal — проект, расширяющий возможности PyTorch Geometric для работы с динамическими графами. Это специализированная библиотека, которая помогает анализировать данные, где связи между объектами меняются со временем: от прогнозирования трафика до эпидемиологических моделей.

Проект имеет:

— Готовые реализации 15+ архитектур из научных статей

— Поддержка временных срезов данных и распределённого обучения через Dask

— Интеграция с PyTorch Lightning для удобной работы с GPU

Для старта достаточно` pip install torch-geometric-temporal `— в комплекте идут датасеты по эпидемиологии, энергетике и веб-трафику.

GitHub (https://github.com/benedekrozemberczki/pytorch_geometric_temporal)

Источник: github.com

Комментарии: