当前位置: 首页 > news >正文

图神经网络DGL框架,graph classification,多个且不同维度的node feature 训练

node feature 维度不同

我现在有许多不同的图要加入训练,每个图的节点特征维度不同,第一张图n_weight特征有10条数据,第二张图n_weight特征有15条数据,但是训练的时候,需要维度都对其,所以直接做0 padding ,把小于15条数据的全部后面填充0,再加入训练

多个node features

我的节点有n_weight和n_community两个特征,都要加入训练
forward 程序中,cat n_weight和n_community两个特征,如果有很多个特征,写循环

class GraphClassifier(nn.Module):def __init__(self, in_dim, hidden_dim, n_classes):super(GraphClassifier, self).__init__()self.conv1 = GraphConv(in_dim, hidden_dim)self.conv2 = GraphConv(hidden_dim, hidden_dim)# flatten into linear so we can crossentropy/ softmax it.self.classify = nn.Linear(hidden_dim, n_classes)def forward(self, g):# run the weight feature through the netw = g.ndata['n_weight']w = F.relu(self.conv1(g, w))w = F.relu(self.conv2(g, w))g.ndata['n_weight'] = w# run the community feature through the netc = g.ndata['n_community']c = F.relu(self.conv1(g, c))c = F.relu(self.conv1(g, c))g.ndata['n_community'] = c# combine both features into one tensorwc = torch.cat((w, c), 1)return self.classify(wc)
http://www.lryc.cn/news/300876.html

相关文章:

  • 蓝桥杯(Web大学组)2022国赛真题:用什么来做计算 A
  • Linux POSIX信号量 线程池
  • Sentinel(理论版)
  • python3 获取某个文件夹所有的pdf文件表格提取表格并一起合并到excel文件
  • 【AIGC】Stable Diffusion的模型入门
  • 【JavaEE】_HTTP请求首行详情
  • Linux第48步_编译正点原子的出厂Linux内核源码
  • 程序员为什么不喜欢关电脑?
  • 【初始RabbitMQ】了解和安装RabbitMQ
  • Linux第56步_根文件系统第3步_将busybox构建的根文件系统烧录到EMMC
  • Linux进程间通信(三)-----System V消息队列
  • Elasticsearch:混合搜索是 GenAI 应用的未来
  • 态、势、感、知的偏序、全序与无序
  • 【从Python基础到深度学习】 8. VIM两种状态
  • java微服务面试篇
  • OpenAI 生成视频模型 Sora 论文翻译
  • 2.13日学习打卡----初学RocketMQ(四)
  • ZigBee学习——BDB
  • 使用Docker快速部署MySQL
  • 力扣热题100_滑动窗口_3_无重复字符的最长子串
  • RM电控工程讲义
  • 论文阅读:《Deep Learning-Based Human Pose Estimation: A Survey》——Part 1:2D HPE
  • C语言——oj刷题——杨氏矩阵
  • C++ 50道面试题
  • 寒假学习记录14:JS字符串
  • 【数学建模】【2024年】【第40届】【MCM/ICM】【C题 网球运动中的“动量”】【解题思路】
  • 无人驾驶LQR控制算法 c++ 实现
  • Karnaugh map (卡诺图)
  • C# CAD 框选pdf输出
  • 【Linux】 Linux 小项目—— 进度条