- 博客(44)
- 收藏
- 关注
原创 异构图上的连接预测二
在这里应该注意到这是GNN网络,用于处理同构图的。也就是边和节点类型都一样的图。# 对子图进行处理咯,例如吧,电影的特征20, 输出为64,# x 的类型被注释为tensor,edge_index 的类型也是tensor,而->tensor 用于指示forward方法的返回类型是tensorreturn x你说分类器干嘛的?假设数据:x_user = [[0.1, 0.2, 0.3], # 用户1的嵌入向量[0.4, 0.5, 0.6], # 用户2的嵌入向量。
2024-05-29 00:13:59
1122
空空如也
空空如也
TA创建的收藏夹 TA关注的收藏夹
TA关注的人