Function getGraph

cogdl/utils/grb_utils.py:27–39

(adj, features: torch.FloatTensor, labels: torch.Tensor = None, device="cpu")

def getGraph(adj, features: torch.FloatTensor, labels: torch.Tensor = None, device="cpu"):
    if type(adj) != torch.Tensor:
        edge_index, edge_attr = adj2edge(adj, device)
        data = Graph(x=features, y=labels, edge_index=edge_index, edge_attr=edge_attr).to(device)
    else:
        if adj.is_sparse:
            adj_np = sp.csr_matrix(adj.to_dense().detach().cpu().numpy())
        else:
            adj_np = sp.csr_matrix(adj.detach().cpu().numpy())
        # print(type(adj_np))
        edge_index, edge_attr = adj2edge(adj_np, device)
        data = Graph(x=features, y=labels, edge_index=edge_index, edge_attr=edge_attr, grb_adj=adj).to(device)
    return data


def updateGraph(graph, adj, features: torch.FloatTensor):