构建自己的图数据集

代码:

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)

相关推荐
IOT.FIVE.NO.19 分钟前
Conda安装pytorch和cuda出现问题的解决记录
人工智能·pytorch·python
苏苏susuus3 小时前
机器学习:load_predict_project
人工智能·机器学习
科技小E3 小时前
打手机检测算法AI智能分析网关V4守护公共/工业/医疗等多场景安全应用
人工智能·安全·智能手机
猿饵块4 小时前
视觉slam--框架
人工智能
yvestine5 小时前
自然语言处理——Transformer
人工智能·深度学习·自然语言处理·transformer
SuperW6 小时前
OPENCV图形计算面积、弧长API讲解(1)
人工智能·opencv·计算机视觉
山海不说话6 小时前
视频行为标注工具BehaviLabel(源码+使用介绍+Windows.Exe版本)
人工智能·python·计算机视觉·视觉检测
虹科数字化与AR7 小时前
安宝特方案丨船舶智造的“AR+AI+作业标准化管理解决方案”(装配)
人工智能·ar·ar眼镜·船舶智造·数字工作流·智能装配
飞哥数智坊8 小时前
Coze实战第13讲:飞书多维表格读取+豆包生图模型,轻松批量生成短剧封面
人工智能
newxtc9 小时前
【配置 YOLOX 用于按目录分类的图片数据集】
人工智能·目标跟踪·分类