graphstorm.inference

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

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

GSgnnNodePredictionInferrer	Inferrer for node prediction tasks.
GSgnnEdgePredictionInferrer	Inferrer for edge prediction tasks.
GSgnnLinkPredictionInferrer	Inferrer for link prediction tasks.
GSgnnEmbGenInferer	Inferrer for embedding generation tasks.