Data Preparation

dataprep-karate.py

import networkx as nx
import numpy as np
import torch

# load graph from networkx library
G = nx.karate_club_graph()

# retrieve the labels for each node
labels = np.asarray([G.nodes[i]['club'] != 'Mr. Hi' for i in G.nodes]).astype(np.int64)

# create edge index from 
adj = nx.to_scipy_sparse_matrix(G).tocoo()
row = torch.from_numpy(adj.row.astype(np.int64)).to(torch.long)
col = torch.from_numpy(adj.col.astype(np.int64)).to(torch.long)
edge_index = torch.stack([row, col], dim=0)

# prepare the embeddings
nodes = pd.DataFrame(list(H.nodes()))
nodes.columns = ['nodes']
nodes['embeddings'] = nodes['nodes'].map(embeddings)
embeddings = torch.from_numpy(np.stack(nodes.embeddings.values))