graphstorm.trainer

GraphStorm training modules assemble the distributed training pipeline for different tasks, e.g., node classification and link prediction.

If possible, users should always use these GraphStorm training modules to avoid handling the complexities involved with the distributed data loading and model training.

Trainers

GSgnnNodePredictionTrainer	Trainer for node prediction tasks.
GSgnnEdgePredictionTrainer	Trainer for edge prediction tasks.
GSgnnLinkPredictionTrainer	Trainer for link prediction tasks.