机器学习笔记
文章平均质量分 63
Coder Bin
这个作者很懒,什么都没留下…
展开
-
图神经网络 torch_geometric 库的 MessagePassing 运行机制学习
torch_geometric.nn.conv. MessagePassing( ) 继承这个类,可以自定义节点信息传播机制 例子 import torch from torch_geometric.nn.conv import MessagePassing class GCNConv(MessagePassing): def __init__(self): #选择相加的方式进行邻居节点信息聚合 super().__init__(aggr='add') def原创 2022-03-17 19:35:57 · 1627 阅读 · 3 评论 -
图神经网络库 torch_geometric 下载
下载地址: 官网下载地址 直接运行官网给出的命令会出现错误: error:Microsoft Visual C++ 14.0 or greater is required. 出现这个错误的原因下载这个安装包需要用到c++编译工具,一种解决方法是直接把需要的关键包下载到本地文件夹再pip install torch-geometric 进入以下网址,找到对应pytorch版本的链接,点进去,选择该版本下对应的torch-geometric所需的关键包,关键包可以都选最新版的下载; 关键包链接 下载完4个关键原创 2022-03-16 17:29:01 · 3032 阅读 · 0 评论 -
Attention机制的总结笔记
人类的视觉注意力 Attention机制借鉴了人类的视觉注意力机制。视觉注意力机制是人类视觉所特有的大脑信号处理机制。人类视觉通过快速扫描全局图像,获得需要重点关注的目标区域,也就是一般所说的注意力焦点,而后对这一区域投入更多注意力资源,以获取更多所需要关注目标的细节信息,而抑制其他无用信息。如下图1:人类用眼睛扫描下面的图片时,红色部分就代表人类在图片中更关注的部分,显然我们把注意力更集中子婴儿的脸部,文本标题,而其它部分信息则不是我们注意力关注的地方。 图1 ...原创 2021-08-28 14:12:56 · 510 阅读 · 0 评论