ggfm.data.construct_graph¶

class ggfm.data.construct_graph(data_dir, graph, src_dst2edge_type)[source]¶

Bases:

Construct dgl.heterograph from ggfm.data.graph.

Parameters:

data_dir (str) – Data directory for saving dgl.heterograph object, which is saved as data_dir/graph.bin.
graph (class:ggfm.data.Graph) – Target graph.
src_dst2edge_type (dict) – The edge types corresponding to (src, dst) types.