(GCN) Graph Convolution Network
input : edges, labels from torch_geometric.nn.models import LightGCN edges = [ [node_id1, node_id2], [node_id1, node_id3], [node_id2, node_id10], ... ] labels = [1,0,0,1,1,1, ...] # for link prediction model = LightGCN(num_nodes=n_node, **kwargs) pred = model(edges) loss = model.link_pred_loss(pred=pred, edge_label=labels) prob = model.predict_link(edge_index=edges, prob=True) acc = accuracy_sco..
2024.01.25