10、图卷积网络与GraphSAGE：产品预测与聚合方法

图卷积网络与GraphSAGE：产品预测与聚合方法

1. 预测消费产品类别

在产品管理中，通过引入产品捆绑销售来提高销售额是一个常见的策略。我们可以使用训练好的模型为给定产品生成产品捆绑组合。

1.1 第一个产品捆绑组合

为了创建产品捆绑组合，需要用到节点嵌入。选择一个产品节点，然后找出与之最相似的六个产品，具体步骤如下：
1. 生成节点嵌入 ：将节点输入训练好的图神经网络（GNN），生成节点嵌入。

gcn_model.eval()
with torch.no_grad():
    gcn_embeddings = gcn_model(subset_graph.x, subset_graph.edge_index, return_embeds=True)

2. 创建相似度矩阵 ：使用余弦相似度比较集合中所有节点的嵌入，创建相似度矩阵。

from sklearn.metrics.pairwise import cosine_similarity
gcn_similarity_matrix = cosine_similarity(gcn_embeddings.cpu().numpy())

3. 列出与所选节点最相似的项目 ：选择一个节点（通过其索引 <