构建自己的图数据集

代码:

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)

相关推荐
小赵AI手记4 分钟前
技术拆解(十五)具身智能:RT-2为什么能听懂“灭绝动物”?VLM知识如何变成动作
人工智能·机器人
就是一顿骚操作5 分钟前
全球 AI 大事件新闻汇总 2026-08-01
大数据·人工智能
阿童木写作25 分钟前
Python实现亚马逊商品图批量智能抠图教程
人工智能·python
EQUINOX126 分钟前
【论文精读】| Qwen-VL
人工智能
小罗水27 分钟前
第14章 Sentinel 限流与 AI 降级
人工智能·sentinel
小弥儿36 分钟前
GitHub今日热榜 | 2026-08-01:硬件安全工具与AI教育并行
人工智能·学习·开源·github
OpenCut38 分钟前
OpenCut丨视频后期剪辑!AI 一键清理视频画面文字水印!
人工智能·音视频
小柯南敲键盘1 小时前
速卖通AI图片翻译API集成实战
人工智能·python
kobesdu1 小时前
流形上的优化:SO(3)与SE(3)的广义加减法在FAST-LIO中如何简化状态估计
人工智能·算法·fastlio
会周易的程序员1 小时前
唯识学架构的AI:Agent个体意识与佛陀出现可能性的推演
人工智能