graphstorm.trainer

GraphStorm trainers assemble the distributed training pipeline for different tasks or different training methods.

If possible, users should always use these trainers to avoid handling the distributed processing and tasks.

Trainers

GSgnnLinkPredictionTrainer

A trainer for link prediction

GSgnnNodePredictionTrainer

A trainer for node prediction

GSgnnEdgePredictionTrainer

Edge prediction trainer.

GLEMNodePredictionTrainer

A trainer for node prediction using GLEM.