dgl 加载bin文件,获得邻接矩阵,不允许使用adj_matrix = graph.adjacency_matrix()
时间: 2024-03-02 08:48:26 浏览: 314
如果您不想使用`adj_matrix = graph.adjacency_matrix()`来获得邻接矩阵,可以使用以下方法:
方法一:使用`dgl.to_networkx`将dgl图对象转换为NetworkX图对象,然后使用NetworkX提供的方法获得邻接矩阵:
```python
import dgl
import networkx as nx
import numpy as np
with open('graph.bin', 'rb') as f:
graph_data = np.load(f, allow_pickle=True)
graph = dgl.graph((graph_data['edge_index'][0], graph_data['edge_index'][1]))
nx_graph = dgl.to_networkx(graph)
adj_matrix = nx.adjacency_matrix(nx_graph)
```
方法二:使用`dgl.to_scipy`将邻接矩阵转换为稀疏矩阵,然后使用`toarray()`方法将稀疏矩阵转换为稠密矩阵:
```python
import dgl
import numpy as np
with open('graph.bin', 'rb') as f:
graph_data = np.load(f, allow_pickle=True)
graph = dgl.graph((graph_data['edge_index'][0], graph_data['edge_index'][1]))
adj_matrix = dgl.to_scipy(graph.adjacency_matrix())
dense_matrix = adj_matrix.toarray()
```
需要注意的是,这两种方法都会将邻接矩阵转换为稠密矩阵,如果邻接矩阵过大,可能会导致内存溢出或性能问题。如果需要处理大规模的邻接矩阵,建议使用稀疏矩阵进行计算。
阅读全文
相关推荐

















