ggfm.data.feature_extractor¶

class ggfm.data.feature_extractor(layer_data, graph)[source]¶

Bases:

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

Extract relevent features.

Parameters:

layer_data (dict) – Sampled node indexes for each node type.
graph (class:ggfm.data.Graph) – Target graph.