![](https://img-blog.csdnimg.cn/20201014180756930.png?x-oss-process=image/resize,m_fixed,h_64,w_64)
图神经网络
文章平均质量分 58
F_aF_a
这个作者很懒,什么都没留下…
展开
-
通过GCN来实现对Cora数据集节点的分类
代码来自《深入浅出图神经网络:GNN原理解析》本节我们通过一个完整的例子来理解 如何通过GCN来实现对节点的分类。我们使用的是Cora数据集,该数据集由2708篇论文,及它们之间的引用关系构成的5429条边组成。这些论文被根据主题划分为7类,分别是神经网络、强化学习、规则学习、概率方法、遗传算法、理论研究、案例相关。每篇论文的特征是通过词袋模型得到的,维度为1433,每-维表示一一个词, 1表示该词在这篇文章中出现过,0表示未出现。首先我们定义类CoraData来对数据进行预处理,主要包括下载数据、原创 2021-11-28 17:27:06 · 3854 阅读 · 0 评论 -
<Transductive Propagation Network for Few-shot Learning>官方代码笔记
文章目录数据模型①`CNNEncoder``RelationNetwork``Prototypical``LabelPropagation`数据模型①CNNEncoder提取特征RelationNetworkPrototypicalLabelPropagation原创 2021-09-11 15:56:09 · 362 阅读 · 6 评论 -
《Few-Shot Learning with Graph Neural Networks》代码理解
代码使用omniglotomniglotomniglot数据集,以5way−1shot5way-1shot5way−1shot为例,一个episodeepisodeepisode只有一张queryqueryquery,一个batchbatchbatch中有300300300个episodeepisodeepisode文章目录数据训练metric_nnWcomputeGconvgmul流程图在main.py中的第144行开始训练的迭代数据首先加载数据# main.py line149data =原创 2021-08-12 21:51:58 · 1561 阅读 · 9 评论 -
《Few-Shot Learning with Graph Neural Networks》
文章目录流程初始化图顶点边图卷积流程初始化图顶点顶点由图的特征和one−hotone-hotone−hot编码构成边两个顶点的边由其顶点的差经过多层感知机MLP得来图卷积原创 2021-08-12 20:32:04 · 168 阅读 · 2 评论 -
anaconda下安装torch_geometric
官方文档安装的时候出了很多错误,最后还是按照官网文档才安装好了文章目录pytorch的版本CUDA的版本安装包pytorch的版本确定pytorch的版本,最低1.4.0python -c "import torch; print(torch.__version__)">>> 1.8.0CUDA的版本确定cuda的版本python -c "import torch; print(torch.version.cuda)">>> 10.2安装包pi原创 2021-08-05 09:41:29 · 1773 阅读 · 2 评论 -
torch_geometric手动下载数据集
程序下载很慢可以复制到浏览器手动下载,然后复制到对应的路径下。我们设置的root = /tmp/ShapeNet此时运行程序,会提示Using exist file shapenetcore_partanno_segmentation_benchmark_v0_normal.zipExtracting /tmp/ShapeNet/shapenetcore_partanno_segmentation_benchmark_v0_normal.zip将压缩包改名为shapenetcore_part原创 2021-08-04 12:09:57 · 780 阅读 · 0 评论 -
PyTorch Geometric的Mini-batches
官方文档 链接文章目录加载ENZYMES数据集batch取出单个数据`ptr`属性`batch`属性加载ENZYMES数据集from torch_geometric.datasets import TUDatasetfrom torch_geometric.data import DataLoaderdataset = TUDataset(root='/tmp/ENZYMES', name='ENZYMES', use_node_attr=True)loader = DataLoader(原创 2021-08-03 20:26:18 · 1499 阅读 · 0 评论