Index

_

__iter__() (graphstorm.dataloading.GSgnnEdgeDataLoader method) (graphstorm.dataloading.GSgnnEdgeDataLoaderBase method) (graphstorm.dataloading.GSgnnLinkPredictionDataLoader method) (graphstorm.dataloading.GSgnnLinkPredictionDataLoaderBase method) (graphstorm.dataloading.GSgnnLinkPredictionTestDataLoader method) (graphstorm.dataloading.GSgnnNodeDataLoader method) (graphstorm.dataloading.GSgnnNodeDataLoaderBase method) (graphstorm.dataloading.GSgnnNodeSemiSupDataLoader method) __len__() (graphstorm.dataloading.GSgnnEdgeDataLoader method) (graphstorm.dataloading.GSgnnEdgeDataLoaderBase method) (graphstorm.dataloading.GSgnnLinkPredictionDataLoader method) (graphstorm.dataloading.GSgnnLinkPredictionDataLoaderBase method) (graphstorm.dataloading.GSgnnLinkPredictionTestDataLoader method) (graphstorm.dataloading.GSgnnNodeDataLoader method) (graphstorm.dataloading.GSgnnNodeDataLoaderBase method) (graphstorm.dataloading.GSgnnNodeSemiSupDataLoader method)

__next__() (graphstorm.dataloading.GSgnnEdgeDataLoader method) (graphstorm.dataloading.GSgnnEdgeDataLoaderBase method) (graphstorm.dataloading.GSgnnLinkPredictionDataLoader method) (graphstorm.dataloading.GSgnnLinkPredictionDataLoaderBase method) (graphstorm.dataloading.GSgnnLinkPredictionTestDataLoader method) (graphstorm.dataloading.GSgnnNodeDataLoader method) (graphstorm.dataloading.GSgnnNodeDataLoaderBase method) (graphstorm.dataloading.GSgnnNodeSemiSupDataLoader method)

B

best_iter_num (graphstorm.eval.GSgnnBaseEvaluator property)

best_test_score (graphstorm.eval.GSgnnBaseEvaluator property) best_val_score (graphstorm.eval.GSgnnBaseEvaluator property)

C

calc_test_scores() (graphstorm.model.LinkPredictDistMultDecoder method) (graphstorm.model.LinkPredictDotDecoder method) compute_score() (graphstorm.eval.GSgnnClassificationEvaluator method) (graphstorm.eval.GSgnnHitsLPEvaluator method) (graphstorm.eval.GSgnnLPRankingEvalInterface method) (graphstorm.eval.GSgnnMrrLPEvaluator method) (graphstorm.eval.GSgnnPerEtypeHitsLPEvaluator method) (graphstorm.eval.GSgnnPerEtypeMrrLPEvaluator method) (graphstorm.eval.GSgnnPredictionEvalInterface method) (graphstorm.eval.GSgnnRconstructFeatRegScoreEvaluator method) (graphstorm.eval.GSgnnRegressionEvaluator method)

create_optimizer() (graphstorm.model.GSgnnModelBase method)

D

data (graphstorm.dataloading.GSgnnEdgeDataLoaderBase property) (graphstorm.dataloading.GSgnnLinkPredictionDataLoaderBase property) (graphstorm.dataloading.GSgnnNodeDataLoaderBase property) decoder_edge_feat_fields (graphstorm.dataloading.GSgnnEdgeDataLoaderBase property)

DenseBiDecoder (class in graphstorm.model) device (graphstorm.model.GSgnnModelBase property) do_early_stop() (graphstorm.eval.GSgnnBaseEvaluator method) do_eval() (graphstorm.eval.GSgnnBaseEvaluator method)

E

edge_feat_field (graphstorm.dataloading.GSgnnData property) edge_feat_fields (graphstorm.dataloading.GSgnnEdgeDataLoaderBase property) (graphstorm.dataloading.GSgnnLinkPredictionDataLoaderBase property) (graphstorm.dataloading.GSgnnNodeDataLoaderBase property) EntityClassifier (class in graphstorm.model) EntityRegression (class in graphstorm.model) eval() (graphstorm.trainer.GSgnnEdgePredictionTrainer method) (graphstorm.trainer.GSgnnLinkPredictionTrainer method) (graphstorm.trainer.GSgnnNodePredictionTrainer method)

eval_frequency (graphstorm.eval.GSgnnBaseEvaluator property) evaluate() (graphstorm.eval.GSgnnClassificationEvaluator method) (graphstorm.eval.GSgnnHitsLPEvaluator method) (graphstorm.eval.GSgnnLPRankingEvalInterface method) (graphstorm.eval.GSgnnMrrLPEvaluator method) (graphstorm.eval.GSgnnPerEtypeHitsLPEvaluator method) (graphstorm.eval.GSgnnPerEtypeMrrLPEvaluator method) (graphstorm.eval.GSgnnPredictionEvalInterface method) (graphstorm.eval.GSgnnRegressionEvaluator method)

F

fanout (graphstorm.dataloading.GSgnnEdgeDataLoaderBase property) (graphstorm.dataloading.GSgnnLinkPredictionDataLoaderBase property) (graphstorm.dataloading.GSgnnNodeDataLoaderBase property) fit() (graphstorm.trainer.GSgnnEdgePredictionTrainer method) (graphstorm.trainer.GSgnnLinkPredictionTrainer method) (graphstorm.trainer.GSgnnNodePredictionTrainer method) forward() (graphstorm.model.DenseBiDecoder method) (graphstorm.model.EntityClassifier method) (graphstorm.model.EntityRegression method) (graphstorm.model.GATConv method) (graphstorm.model.GATEncoder method) (graphstorm.model.GATv2Conv method) (graphstorm.model.GATv2Encoder method) (graphstorm.model.GSgnnEdgeModelInterface method) (graphstorm.model.GSgnnLinkPredictionModelInterface method) (graphstorm.model.GSgnnNodeModelInterface method) (graphstorm.model.GSLMNodeEncoderInputLayer method) (graphstorm.model.GSNodeEncoderInputLayer method) (graphstorm.model.GSPureLMNodeInputLayer method) (graphstorm.model.HGTEncoder method) (graphstorm.model.HGTLayer method) (graphstorm.model.LinkPredictContrastiveDistMultDecoder method) (graphstorm.model.LinkPredictContrastiveDotDecoder method) (graphstorm.model.LinkPredictDistMultDecoder method) (graphstorm.model.LinkPredictDotDecoder method) (graphstorm.model.MLPEdgeDecoder method) (graphstorm.model.MLPEFeatEdgeDecoder method) (graphstorm.model.RelationalAttLayer method) (graphstorm.model.RelationalGATEncoder method) (graphstorm.model.RelationalGCNEncoder method) (graphstorm.model.RelGraphConvLayer method) (graphstorm.model.SAGEConv method) (graphstorm.model.SAGEEncoder method)

freeze() (graphstorm.model.GSLMNodeEncoderInputLayer method) (graphstorm.model.GSPureLMNodeInputLayer method)

G

g (graphstorm.dataloading.GSgnnData property) GATConv (class in graphstorm.model) GATEncoder (class in graphstorm.model) GATv2Conv (class in graphstorm.model) GATv2Encoder (class in graphstorm.model) get_edge_feats() (graphstorm.dataloading.GSgnnData method) get_edge_infer_set() (graphstorm.dataloading.GSgnnData method) get_edge_test_set() (graphstorm.dataloading.GSgnnData method) get_edge_train_set() (graphstorm.dataloading.GSgnnData method) get_edge_val_set() (graphstorm.dataloading.GSgnnData method) get_general_dense_parameters() (graphstorm.model.GSLMNodeEncoderInputLayer method) (graphstorm.model.GSPureLMNodeInputLayer method) get_lm_dense_parameters() (graphstorm.model.GSLMNodeEncoderInputLayer method) (graphstorm.model.GSPureLMNodeInputLayer method) get_node_feat_size() (in module graphstorm.gsf) get_node_feats() (graphstorm.dataloading.GSgnnData method) get_node_infer_set() (graphstorm.dataloading.GSgnnData method) get_node_test_set() (graphstorm.dataloading.GSgnnData method) get_node_train_set() (graphstorm.dataloading.GSgnnData method) get_node_val_set() (graphstorm.dataloading.GSgnnData method) get_rel_names_for_reconstruct() (in module graphstorm.gsf) get_relemb() (graphstorm.model.LinkPredictDistMultDecoder method) get_relembs() (graphstorm.model.LinkPredictDistMultDecoder method) get_sparse_params() (graphstorm.model.GSNodeEncoderInputLayer method) get_unlabeled_node_set() (graphstorm.dataloading.GSgnnData method) get_val_score_rank() (graphstorm.eval.GSgnnBaseEvaluator method) (graphstorm.eval.GSgnnPerEtypeHitsLPEvaluator method) (graphstorm.eval.GSgnnPerEtypeMrrLPEvaluator method) graph_name (graphstorm.dataloading.GSgnnData property) GSgnnBaseEvaluator (class in graphstorm.eval) GSgnnClassificationEvaluator (class in graphstorm.eval) GSgnnData (class in graphstorm.dataloading) GSgnnEdgeDataLoader (class in graphstorm.dataloading)

GSgnnEdgeDataLoaderBase (class in graphstorm.dataloading) GSgnnEdgeModelBase (class in graphstorm.model) GSgnnEdgeModelInterface (class in graphstorm.model) GSgnnEdgePredictionInferrer (class in graphstorm.inference) GSgnnEdgePredictionTrainer (class in graphstorm.trainer) GSgnnEmbGenInferer (class in graphstorm.inference) GSgnnHitsLPEvaluator (class in graphstorm.eval) GSgnnLinkPredictionDataLoader (class in graphstorm.dataloading) GSgnnLinkPredictionDataLoaderBase (class in graphstorm.dataloading) GSgnnLinkPredictionInferrer (class in graphstorm.inference) GSgnnLinkPredictionModelBase (class in graphstorm.model) GSgnnLinkPredictionModelInterface (class in graphstorm.model) GSgnnLinkPredictionPredefinedTestDataLoader (class in graphstorm.dataloading) GSgnnLinkPredictionTestDataLoader (class in graphstorm.dataloading) GSgnnLinkPredictionTrainer (class in graphstorm.trainer) GSgnnLPRankingEvalInterface (class in graphstorm.eval) GSgnnModelBase (class in graphstorm.model) GSgnnMrrLPEvaluator (class in graphstorm.eval) GSgnnNodeDataLoader (class in graphstorm.dataloading) GSgnnNodeDataLoaderBase (class in graphstorm.dataloading) GSgnnNodeModelBase (class in graphstorm.model) GSgnnNodeModelInterface (class in graphstorm.model) GSgnnNodePredictionInferrer (class in graphstorm.inference) GSgnnNodePredictionTrainer (class in graphstorm.trainer) GSgnnNodeSemiSupDataLoader (class in graphstorm.dataloading) GSgnnPerEtypeHitsLPEvaluator (class in graphstorm.eval) GSgnnPerEtypeMrrLPEvaluator (class in graphstorm.eval) GSgnnPredictionEvalInterface (class in graphstorm.eval) GSgnnRconstructFeatRegScoreEvaluator (class in graphstorm.eval) GSgnnRegressionEvaluator (class in graphstorm.eval) GSLMNodeEncoderInputLayer (class in graphstorm.model) GSNodeEncoderInputLayer (class in graphstorm.model) GSPureLMNodeInputLayer (class in graphstorm.model)

H

has_edge_feats() (graphstorm.dataloading.GSgnnData method) has_edge_lm_feats() (graphstorm.dataloading.GSgnnData method) has_node_feats() (graphstorm.dataloading.GSgnnData method)

has_node_lm_feats() (graphstorm.dataloading.GSgnnData method) HGTEncoder (class in graphstorm.model) HGTLayer (class in graphstorm.model) history (graphstorm.eval.GSgnnBaseEvaluator property)

I

in_dims (graphstorm.model.DenseBiDecoder property) (graphstorm.model.EntityClassifier property) (graphstorm.model.EntityRegression property) (graphstorm.model.GSNodeEncoderInputLayer property) (graphstorm.model.GSPureLMNodeInputLayer property) (graphstorm.model.LinkPredictDistMultDecoder property) (graphstorm.model.LinkPredictDotDecoder property) (graphstorm.model.MLPEdgeDecoder property)

infer() (graphstorm.inference.GSgnnEdgePredictionInferrer method) (graphstorm.inference.GSgnnEmbGenInferer method) (graphstorm.inference.GSgnnLinkPredictionInferrer method) (graphstorm.inference.GSgnnNodePredictionInferrer method) initialize() (in module graphstorm.gsf)

L

label_field (graphstorm.dataloading.GSgnnEdgeDataLoaderBase property) (graphstorm.dataloading.GSgnnNodeDataLoaderBase property) LinkPredictContrastiveDistMultDecoder (class in graphstorm.model)

LinkPredictContrastiveDotDecoder (class in graphstorm.model) LinkPredictDistMultDecoder (class in graphstorm.model) LinkPredictDotDecoder (class in graphstorm.model)

M

metric_list (graphstorm.eval.GSgnnBaseEvaluator property) MLPEdgeDecoder (class in graphstorm.model)

MLPEFeatEdgeDecoder (class in graphstorm.model) multilabel (graphstorm.eval.GSgnnClassificationEvaluator property)

N

node_feat_field (graphstorm.dataloading.GSgnnData property) node_feat_fields (graphstorm.dataloading.GSgnnEdgeDataLoaderBase property) (graphstorm.dataloading.GSgnnLinkPredictionDataLoaderBase property) (graphstorm.dataloading.GSgnnNodeDataLoaderBase property)

O

out_dims (graphstorm.model.DenseBiDecoder property) (graphstorm.model.EntityClassifier property) (graphstorm.model.EntityRegression property) (graphstorm.model.GSNodeEncoderInputLayer property) (graphstorm.model.GSPureLMNodeInputLayer property) (graphstorm.model.LinkPredictDistMultDecoder property) (graphstorm.model.LinkPredictDotDecoder property) (graphstorm.model.MLPEdgeDecoder property)

P

pos_graph_edge_feat_fields (graphstorm.dataloading.GSgnnLinkPredictionDataLoaderBase property) predict() (graphstorm.model.DenseBiDecoder method) (graphstorm.model.EntityClassifier method) (graphstorm.model.EntityRegression method) (graphstorm.model.GSgnnEdgeModelInterface method) (graphstorm.model.GSgnnNodeModelInterface method) (graphstorm.model.MLPEdgeDecoder method) (graphstorm.model.MLPEFeatEdgeDecoder method)

predict_proba() (graphstorm.model.DenseBiDecoder method) (graphstorm.model.EntityClassifier method) (graphstorm.model.EntityRegression method) (graphstorm.model.MLPEdgeDecoder method) (graphstorm.model.MLPEFeatEdgeDecoder method) prepare() (graphstorm.model.GSLMNodeEncoderInputLayer method) (graphstorm.model.GSPureLMNodeInputLayer method) prepare_input_encoder() (graphstorm.model.GSgnnModelBase method)

R

RelationalAttLayer (class in graphstorm.model) RelationalGATEncoder (class in graphstorm.model) RelationalGCNEncoder (class in graphstorm.model) RelGraphConvLayer (class in graphstorm.model) require_cache_embed() (graphstorm.model.GSLMNodeEncoderInputLayer method) (graphstorm.model.GSNodeEncoderInputLayer method) (graphstorm.model.GSPureLMNodeInputLayer method)

restore_dense_model() (graphstorm.model.GSgnnModelBase method) restore_model() (graphstorm.model.GSgnnModelBase method) restore_sparse_model() (graphstorm.model.GSgnnModelBase method)

S

SAGEConv (class in graphstorm.model) SAGEEncoder (class in graphstorm.model) save_dense_model() (graphstorm.model.GSgnnModelBase method)

save_model() (graphstorm.model.GSgnnModelBase method) save_sparse_model() (graphstorm.model.GSgnnModelBase method) setup_task_tracker() (graphstorm.eval.GSgnnBaseEvaluator method)

T

target_eidx (graphstorm.dataloading.GSgnnEdgeDataLoaderBase property) (graphstorm.dataloading.GSgnnLinkPredictionDataLoaderBase property)

target_nidx (graphstorm.dataloading.GSgnnNodeDataLoaderBase property) task_tracker (graphstorm.eval.GSgnnBaseEvaluator property)

U

unfreeze() (graphstorm.model.GSLMNodeEncoderInputLayer method) (graphstorm.model.GSPureLMNodeInputLayer method)

use_wholegraph_sparse_emb (graphstorm.model.GSNodeEncoderInputLayer property)

V

val_perf_rank_list (graphstorm.eval.GSgnnBaseEvaluator property)