Welcome to DeepGNN’s documentation!¶
DeepGNN is a package for training/evaluating ML models on graph data. It is a Python library that provides:
- A graph engine object designed for ML tasks with an assortment of routines for sampling nodes, edges and neighbors as well as feature fetching.
- Various aggregators, encoders and decoders to pass graph data to neural nets.
- Basic NN layers for training: convolution, attention and bindings to pytorch-geometric library.
- A collection of trainers to work with models in local and distributed environments.