随笔分类 -  图神经网络

7、生成mask，过滤无效张量
摘要：1、准备环境 import torch random_tensor = torch.randn(10, 2) print(random_tensor) 2、准备batch索引 from torch_geometric.utils import to_dense_batch, to_dense_adj 阅读全文
posted @ 2023-10-24 22:59 jasonzhangxianrong 阅读(53) 评论(0) 推荐(0) 编辑
5、SimGNN实战
摘要：一、概述 文献标题：SimGNN： A Neural Network Approach to Fast Graph Similarity Computation来源：WSDM2018（ 网络搜索和数据挖掘国际会议）论文链接：https://arxiv.org/abs/1808.05689代码链接：h 阅读全文
posted @ 2023-09-26 22:33 jasonzhangxianrong 阅读(496) 评论(2) 推荐(0) 编辑
4、scatter作用
摘要：scatter_add是torch_scatter库中的一个函数，用于对输入张量进行聚合操作，并将聚合结果累加到指定位置上。 具体来说，scatter_add函数的使用方法如下： from torch_scatter import scatter_add # 定义输入张量 input_tensor 阅读全文
posted @ 2023-09-26 20:36 jasonzhangxianrong 阅读(189) 评论(0) 推荐(0) 编辑
3、自创建数据集：基于点击率预测
摘要：1、如何制作自己的图数据 import warnings warnings.filterwarnings("ignore") import torch 创建一个图，信息如下： x是每个点的输入特征，y是每个点的标签 x = torch.tensor([[2,1], [5,6], [3,7], [12 阅读全文
posted @ 2023-09-25 22:21 jasonzhangxianrong 阅读(236) 评论(0) 推荐(0) 编辑
2、点分类任务
摘要：1、Cora dataset(数据集描述：Yang et al. (2016)) 论文引用数据集，每一个点有1433维向量 最终要对每个点进行7分类任务(每个类别只有20个点有标注) from torch_geometric.datasets import Planetoid#下载数据集用的 fro 阅读全文
posted @ 2023-09-25 20:39 jasonzhangxianrong 阅读(173) 评论(0) 推荐(0) 编辑
1、pytorch_geometric基本使用
摘要：1、工具包安装方法： 一定参考其GITHUB：https://github.com/pyg-team/pytorch_geometric (千万不要pip直接安装，肯定不行的) （1）先安装编译好的包： https://data.pyg.org/whl/ （2）再安装整体 pip install t 阅读全文
posted @ 2023-09-25 19:36 jasonzhangxianrong 阅读(659) 评论(0) 推荐(1) 编辑