您当前的位置: 首页 > 

静静喜欢大白

暂无认证

  • 2浏览

    0关注

    521博文

    0收益

  • 0浏览

    0点赞

    0打赏

    0留言

私信
关注
热门博文

PyG-数据下载(内置函数)和数据属性信息统计

静静喜欢大白 发布时间:2021-02-10 10:42:45 ,浏览量:2

1、简介 数据集下载路径问题
dataset = Planetoid(root='./data/',name='Cora')  # ./下载到当前同级下的data文件下(自动创建data目录,具体的Cora下载就是内置函数的事情了)

dataset = TUDataset(root='/data/ENZYMES', name='ENZYMES')  # /是直接在F盘下创建data/ENZYMES路径,并使用内置函数处理下载ENZYMES数据
2、代码
'''
PyG手册入门

'''


# 给出了一个具有3个节点和4条边的无权无向图的简单例子,每个节点只包含一个特征

import torch  # 基于pytorch的
from torch_geometric.data import Data  # 数据处理接口,一般为了初始化节点和邻接矩阵

# 对于一个具有3个节点,4条边的具体简单无权无向图(边的2倍,正反)
'''
需要初始化节点特征x【N,F】和邻接矩阵edge_index【2,E】
    N节点数
    F节点特征维度数
    E边数


注意
X
    在这里节点特征x就是节点本身的值,维度就是1了
    数据类型:long
edge_index
    即定义所有边的源节点和目标节点的张量,并不是索引元组的列表
    数据类型:float
   

'''
#边,边结构信息初始化如下;shape = [2,num_edge]
edge_index = torch.tensor([[0,
关注
打赏
1510642601
查看更多评论
立即登录/注册

微信扫码登录

0.0386s