构建自己的图数据集

代码:

python 复制代码
import warnings
warnings.filterwarnings("ignore")
import torch
from torch_geometric.data import Data

x = torch.tensor([[2,1],[5,6],[3,7],[12,0]],dtype=torch.float)
y = torch.tensor([0,1,0,1],dtype=torch.float)

#定义边
edge_index = torch.tensor([[0,1,2,0,3], #起始点
                           [1,0,1,3,2]],dtype=torch.long) #终止点


data = Data(x=x,edge_index=edge_index,y=y)
print(data)

TopKPooling流程:

对图片进行剪枝操作,选择分低的节点剔除掉,然后再重新组合成一个新的图

图注意力机制与序列图模型:

序列图神经网络(GNN)

相关推荐
珂朵莉MM1 分钟前
第七届全球校园人工智能算法精英大赛-算法巅峰赛产业命题赛第3赛季优化题--束搜索
人工智能·算法
智慧物业老杨1 分钟前
智慧物业合同周期管理系统:从风险预警到智能交接的全流程数智化落地方案
java·人工智能·python
科技AI训练师1 分钟前
2026高压清洗泵厂家选择指南:判断标准与选购要点
大数据·人工智能
Front思3 分钟前
AI前端工程师需要具备能力+
前端·人工智能·ai
Percent_bigdata4 分钟前
“模数共振”开启产业AI新阶段,重新定义数据治理
大数据·人工智能
AI品信智慧数智人7 分钟前
告别传统导游!伴游小助手,AI智能解锁全新旅行体验✨
人工智能·旅游
乐维_lwops15 分钟前
从 “救火运维” 到 “自动驾驶”:运维智能体到底解决了什么?
运维·人工智能·运维智能体
TheRouter37 分钟前
AI Agent 记忆体系建设实战:短期、长期与工作记忆的工程实现
数据库·人工智能·oracle
weixin_4684668541 分钟前
MoneyPrinterTurbo 短视频自动化生产实战指南
运维·人工智能·自动化·大模型·音视频·moneyprinter
Omics Pro1 小时前
首个!外源天然产物综合性代谢图谱
数据库·人工智能·算法·机器学习·r语言