推荐算法
文章平均质量分 76
sweet_Mary
这个作者很懒,什么都没留下…
展开
-
KGCN---pytorch代码(3)---model
KGCN原创 2024-03-18 17:24:15 · 846 阅读 · 0 评论 -
KGCN---pytorch代码(2)---aggregator
将当前节点的向量(self_vectors)与邻居节点的向量(neighbor_vectors)+邻居关系(neighbor_relations)+以及用户嵌入(user_embeddings)+act(激活函数)结合。在得分张量的最后添加一个维度,将其形状从 [batch_size, -1, n_neighbor] 调整为 [batch_size, -1, n_neighbor, 1]结果是一个形状为 [batch_size, -1, n_neighbor] 的张量,表示每个邻居对当前节点的重要性得分。原创 2024-03-16 20:00:45 · 407 阅读 · 0 评论 -
KGCN---pytorch代码(1)---data_loader
df_rating:把data/music/user_artists.dat读入,['userID', 'itemID', 'rating'],只保留在 df_item2id中存在的itemID行,并重置索引。df_kg:把data/music/kg.txt读入,['head','relation','tail']df_item2id:把item_index2entity_id.txt读入,['item','id']获取每个编码器编码类别数量。原创 2024-03-08 21:14:03 · 565 阅读 · 0 评论