构建自己的图数据集

代码:

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)

相关推荐
说私域1 分钟前
基于开源AI智能名片链动2+1模式S2B2C商城小程序的移动互联网人气氛围营造机制研究
人工智能·小程序·开源
TMT星球9 分钟前
从IFA再出发:中国制造与海信三筒洗衣机的“答案”
人工智能·制造
edisao11 分钟前
[特殊字符] 从助手到引擎:基于 GPT 的战略协作系统演示
大数据·人工智能·gpt
三之又三1 小时前
卷积神经网络CNN-part5-NiN
人工智能·神经网络·cnn
百锦再1 小时前
在 CentOS 系统上实现定时执行 Python 邮件发送任务
java·linux·开发语言·人工智能·python·centos·pygame
Amy187021118231 小时前
中线安防保护器,也叫终端电气综合治理保护设备为现代生活筑起安全防线
人工智能·安全·智慧城市
CV-杨帆1 小时前
论文阅读:ACL 2024 Stealthy Attack on Large Language Model based Recommendation
论文阅读·人工智能·语言模型
飞哥数智坊1 小时前
AI 编程太混乱?我的3个实践,防止代码失控
人工智能·ai编程
NMGWAP1 小时前
AI辅助编程:软件工程的终结还是进化新阶段?
人工智能·软件工程
云边云科技2 小时前
企业跨区域组网新解:SD-WAN技术打造安全稳定网络体系
运维·网络·人工智能·安全·边缘计算