API Reference
The released APIs include the necessary components that can build GraphStorm models, training and inference pipelines. For examples about how to use these APIs, please refer to the API Programming Examples.
Note
This API reference does not include the components for multi-task learning, which is still an experimental feature of Graphstorm v0.3, and may need extensive modification.
- graphstorm
- graphstorm.config
- graphstorm.dataloading.dataset
- graphstorm.dataloading.dataloading
- graphstorm.eval
- graphstorm.inference
- graphstorm.model
- GSgnnModelBase
- GSgnnNodeModelBase
- GSgnnNodeModelInterface
- GSgnnEdgeModelBase
- GSgnnEdgeModelInterface
- GSgnnLinkPredictionModelBase
- GSgnnLinkPredictionModelInterface
- GSPureLearnableInputLayer
- GSNodeEncoderInputLayer
- GSLMNodeEncoderInputLayer
- GSPureLMNodeInputLayer
- RelGraphConvLayer
- RelationalGCNEncoder
- RelationalAttLayer
- RelationalGATEncoder
- HGTLayer
- HGTLayerwithEdgeFeat
- HGTEncoder
- SAGEConv
- SAGEEncoder
- GATConv
- GATEncoder
- GATv2Conv
- GATv2Encoder
- GATConvwithEdgeFeat
- EntityClassifier
- EntityRegression
- DenseBiDecoder
- MLPEdgeDecoder
- MLPEFeatEdgeDecoder
- EdgeRegression
- LinkPredictDotDecoder
- LinkPredictContrastiveDotDecoder
- LinkPredictDistMultDecoder
- LinkPredictContrastiveDistMultDecoder
- LinkPredictRotatEDecoder
- LinkPredictContrastiveRotatEDecoder
- LinkPredictWeightedRotatEDecoder
- LinkPredictTransEDecoder
- LinkPredictContrastiveTransEDecoder
- LinkPredictWeightedTransEDecoder
- ClassifyLossFunc
- FocalLossFunc
- RegressionLossFunc
- ShrinkageLossFunc
- LinkPredictBCELossFunc
- WeightedLinkPredictBCELossFunc
- LinkPredictAdvBCELossFunc
- WeightedLinkPredictAdvBCELossFunc
- LinkPredictContrastiveLossFunc
- graphstorm.trainer