graphstorm.gsf.get_rel_names_for_reconstruct

graphstorm.gsf.get_rel_names_for_reconstruct(g, reconstructed_embed_ntype, feat_size)

Get the edge type list for reconstructing node features.

Parameters

gDistGraph

A DGL distributed graph.

reconstructed_embed_ntypelist of str

The node types for which node features need to be reconstructed.

feat_sizedict of int

The feature size on each node type in the format of {“ntype”: size}.

Returns

reconstruct_etypes: list of tuples

The edge types whose destination nodes required for reconstructing node features.