加入收藏 | 设为首页 | 会员中心 | 我要投稿 网站开发网_马鞍山站长网 (https://www.0555zz.com/)- 科技、建站、经验、云计算、5G、大数据,站长网!
当前位置: 首页 > 建站 > 正文

图卷积网络到底怎么做,这是一份极简的Numpy实现

发布时间:2019-02-22 06:51:05 所属栏目:建站 来源:机器之心编译
导读:副标题#e# 由于图结构非常复杂且信息量很大,因此对于图的机器学习是一项艰巨的任务。本文介绍了如何使用图卷积网络(GCN)对图进行深度学习,GCN 是一种可直接作用于图并利用其结构信息的强大神经网络。 本文将介绍 GCN,并使用代码示例说明信息是如何通过 G

接下来,我们将随机初始化权重。

  1. W_1 = np.random.normal( 
  2.     loc=0, scale=1, size=(zkc.number_of_nodes(), 4)) 
  3. W_2 = np.random.normal( 
  4.     loc=0, size=(W_1.shape[1], 2)) 

接着,我们会堆叠 GCN 层。这里,我们只使用单位矩阵作为特征表征,即每个节点被表示为一个 one-hot 编码的类别变量。

  1. def gcn_layer(A_hat, D_hat, X, W): 
  2.     return relu(D_hat**-1 * A_hat * X * W) 
  3. H_1 = gcn_layer(A_hat, D_hat, I, W_1) 
  4. H_2 = gcn_layer(A_hat, D_hat, H_1, W_2) 
  5. output = H_2 

(编辑:网站开发网_马鞍山站长网)

【声明】本站内容均来自网络,其相关言论仅代表作者个人观点,不代表本站立场。若无意侵犯到您的权利,请及时与联系站长删除相关内容!