torch_geometric的同构图,异构图,超图

torch_geometric的同构图,异构图,超图

1. 异构图

  1. 构建异构图

导入库

from torch_geometric.data import HeteroData, Batch

实例化一个图

graph_text = HeteroData()
# 初始化 结点类型
graph_text[‘xx’], graph_text[‘xx’], graph_text[‘xx’]
# 初始化 边类型
graph_text[“xx”, “connect”, “xxx”]

如果使用 batch

---- 组成batch
batch_graph = [g1, g2, …]
g_batch = Batch.from_data_list(batch_graph)
---- 拆开
cur_audio_graph_list = cur_audio_graph_batch.to_data_list()

  1. 构建异构图网络

self.heteroConv1 = HeteroConv({
(“xxxx”, “connect”, “xxxx”): SAGEConv(hidden_channels, hidden_channels),

}, aggr=‘sum’)

  1. 使用异构图

enc = self.heteroConv1(graph_text .x_dict, graph_text .edge_index_dict)

2. 同构图

同异构图一样, 主要就是 组成数据的时候

3. 超图

待更新

评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值