ggfm.data

Graph

ggfm.data.Graph

The Heterogeneous Graph Transformer (HGT) operator from the "Heterogeneous Graph Transformer" paper.

ggfm.data.HomogeneousGraph

Graph Preprocessing

ggfm.data.renamed_load

Load Graph object.

ggfm.data.sample_subgraph

"GPT-GNN: Generative Pre-Training of Graph Neural Networks" paper.

ggfm.data.get_type_id

Statistically analyze the type_degin_ids and graph_node_type of graphs.

ggfm.data.construct_graph

Construct dgl.heterograph from ggfm.data.graph.

ggfm.data.generate_lm_embs

Language model generates node embeddings for words。

ggfm.data.feature_extractor

"GPT-GNN: Generative Pre-Training of Graph Neural Networks" paper.

ggfm.data.higpt_prompt_generation

Main function to execute all dataset processing steps.

ggfm.data.construct_link_and_node

Construct link.dat and node.dat.

ggfm.data.construct_graph_node_name

Construct graph_node_name.pkl for ggfm.data.graph.

ggfm.data.metapath_based_corpus_construction

Metapath-based corpus construction.

ggfm.data.random_walk_based_corpus_construction

Construct link.dat and node.dat.

ggfm.data.read_npz

Read a npz file.

ggfm.data.extract_zip

Extracts a zip archive to a specific folder.

ggfm.data.download_url

Downloads the content of an URL to a specific folder.

ggfm.data.download_google_url

Downloads the content of a Google Drive ID to a specific folder.

ggfm.data.args_print

Print argments.

ggfm.data.open_pkl_file

Open pickle file.

ggfm.data.open_txt_file

Open txt file.

ggfm.data.save_pkl_file

Save pickle file.

ggfm.data.save_txt_file

Save txt file.

ggfm.data.get_train_val_test_split

Split the dataset into train, validation, and test sets.

ggfm.data.generate_masks