Node Classification with Graph Neural Networks
keras.io › examples › graph# create an edges array (sparse adjacency matrix) of shape [2, num_edges]. edges = citations[ ["source", "target"]].to_numpy().t # create an edge weights array of ones. edge_weights = tf.ones(shape=edges.shape[1]) # create a node features array of shape [num_nodes, num_features]. node_features = tf.cast( papers.sort_values("paper_id") …