dgl 加载bin文件,获得邻接矩阵,不允许使用adj_matrix = graph.adjacency_matrix().to_dense()
时间: 2024-03-02 08:48:25 浏览: 96
如果您不想使用`adj_matrix = graph.adjacency_matrix().to_dense()`来获得邻接矩阵,可以使用以下方法:
方法一:使用`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()
```
方法二:使用`dgl.backend.numpy()`将邻接矩阵转换为numpy数组,然后使用`numpy.ndarray`的`tolist()`方法将numpy数组转换为Python列表:
```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 = graph.adjacency_matrix()
dense_matrix = dgl.backend.numpy()((adj_matrix != 0).astype(np.int32)).tolist()
```
需要注意的是,这两种方法都会将邻接矩阵转换为稠密矩阵,如果邻接矩阵过大,可能会导致内存溢出或性能问题。如果需要处理大规模的邻接矩阵,建议使用稀疏矩阵进行计算。
阅读全文