使用dgl展示Cora数据集
时间: 2023-07-09 16:08:13 浏览: 263
core data demo
4星 · 用户满意度95%
首先,你需要安装dgl库。可以通过以下命令进行安装:
```bash
pip install dgl
```
然后,你需要下载Cora数据集。可以使用以下命令进行下载:
```bash
wget https://data.dgl.ai/dataset/cora_raw.zip
unzip cora_raw.zip
```
接下来,你可以使用以下代码来加载数据集并使用dgl构建图:
```python
import dgl
import numpy as np
import pandas as pd
import torch
from dgl.data import citation_graph as citegrh
# 加载数据集
data = citegrh.load_cora()
# 创建图
g = dgl.DGLGraph()
g.add_nodes(data.num_nodes)
g.add_edges(data.edge_index[0], data.edge_index[1])
# 设置节点特征
features = torch.FloatTensor(data.features)
g.ndata['feat'] = features
# 设置标签
labels = torch.LongTensor(data.labels)
g.ndata['label'] = labels
# 打印图的信息
print(g)
```
这样就可以加载Cora数据集并使用dgl来构建图了。
阅读全文