config

Classes that define configuration options for training or using a GNN Model.

Modules

data

Config classes for defining datasets.

model

Config classes for defining a GNNModel

optimizer

Config classes for defining optimizers

training

The TrainingConfig class

Classes

DataConfig

A config class for setting up training, validation, and test datasets.

DatasetConfig

A config class for a single dataset.

ModelConfig

The configuration class for a GNNModel

OptimizerConfig

The configuration for the optimizer to use during training

TrainingConfig

Overall config class for training a model