DGL阅读笔记
文章平均质量分 96
正经放
这个作者很懒,什么都没留下…
展开
-
dgl源码阅读笔记(3)——DeepWalk
dgl源码阅读笔记(3)——DeepWalk图神经网络开源库dgl阅读笔记文章目录dgl源码阅读笔记(3)——DeepWalk图神经网络开源库dgl阅读笔记@[TOC](文章目录)前言一、DeepWalk简单回顾二、浏览代码类1.class DeepwalkTrainer2.class SkipGramModel.__init__()def init_emb2pos_indexdef init_emb2neg_indexclass SkipGramModel的其他函数def fast_learn三原创 2022-03-29 21:40:06 · 1215 阅读 · 0 评论 -
dgl源码阅读笔记(2)——GCN
注意到sample中walks取[0],对应的就是traces,所以walks其实就是返回了一个游走节点序列。关于SkipGram这里的参数每一个都先存到self的同名变量中,就不截图了。接下来首先是对u和v的embedding进行随机初始化并把embedding限制在initrange(1/128)的正态分布内再用0填充weight中的空值,虽然我怀疑是否用得上接下来实现lookuptable的功能,即不用乘法,而是利用下标的方式取出对应向量首先对-6.01到6.01中间隔0.01的每一个元原创 2022-03-26 19:52:37 · 5319 阅读 · 3 评论 -
dgl源码阅读笔记(1)——update_all
dgl源码阅读笔记(1)——update_all图神经网络开源库dgl阅读笔记文章目录dgl源码阅读笔记(1)——update_all前言一、update_all1、is_builtin2、message_passing3、invoke_gspmm4、_set_n_repr5、update前言update_all是dgl实现消息传递的重要函数,这里进行阅读了解在debug过程是运行GCN模型后引用的该函数,有些数据集特征数,比如2708等,都是受数据集影响的,在文中为了方便理解所以写了下来。原创 2022-03-26 19:50:26 · 2544 阅读 · 1 评论
分享