首页 > 解决方案 > 使用 python 或 R 将邻接矩阵转换为 torch_geometric.data.Data 格式

问题描述

我正在尝试将邻接矩阵转换为 torch_geometric.data.Data 格式。我能够使用 csr_matrix 边缘索引列表。

我也想知道我应该为 x: Node feature matrix with shape [num_nodes, num_node_features] 放什么,这是否应该是边缘权重的矩阵?最好能清楚地了解节点特征列与什么相关;具有理论应用的实际或理论示例会很棒。

帮助将不胜感激。

import torch

# creating tensor from targets_df 
torch_tensor = torch.tensor(adjacencyMat_df.iloc[: , 1:].values) #https://stackoverflow.com/questions/50307707/convert-pandas-dataframe-to-pytorch-tensor
torch_tensor

在此处输入图像描述

import scipy.sparse as scpy

A=scpy.csr_matrix(torch_tensor)
print(A)

data = Data(x=x, edge_index=edge_index)

标签: pythonrgraphadjacency-matrixpytorch-dataloader

解决方案


和我最初想的一样。它可能涉及所选特征的 90 度旋转,每个样本结果沿列。然后每一行代表一个节点。


推荐阅读