python - 使用 TensorFlow 的推荐系统 (SVD)

我正在尝试创建一个协同过滤算法来向某些用户推荐产品。

我很快就开始使用 TensorFlow(我认为它足够有效和灵活)。
我发现这段代码可以做我感兴趣的事情，创建模型并训练用户 ID、产品和评级:https://github.com/songgc/TF-recomm

我启动了代码并训练了模型。

训练模型后，我需要进行预测，即为每个用户获取建议，以便将它们保存在我使用 NODE.js 应用程序访问的数据库中。

培训完成后，如何为每个用户检索此建议列表？

if __name__ == '__main__':
    df_train, df_test=get_data()
    svd(df_train, df_test)
    print("Done!")

最佳答案

需要修改代码的预测部分，输出top K推荐产品。进行预测的当前代码是:

 embd_user = tf.nn.embedding_lookup(w_user, user_batch, name="embedding_user")
 embd_item = tf.nn.embedding_lookup(w_item, item_batch, name="embedding_item")
 infer = tf.reduce_sum(tf.multiply(embd_user, embd_item), 1)

这里 embed_user 是特定用户的用户嵌入，embd_item 是特定项目的。因此，不是将 particular user 与 particular item 进行比较，而是需要更改它以将其与所有项目进行比较。矩阵 w_item 是所有项目的嵌入。这可以通过以下方式完成:

 embd_user = tf.nn.embedding_lookup(w_user, user_batch, name="embedding_user")
 # Multiply user embedding of shape: [1 x dim]
 # with every item embeddings of shape: [item_num, dim],
 # to produce rank of all items of shape: [item_num]
 predict = tf.matmul(embd_user, w_item, transpose_b=True)

然后可以选择预测输出中最大值的top k索引。

关于python - 使用 TensorFlow 的推荐系统 (SVD)，我们在Stack Overflow上找到一个类似的问题：https://stackoverflow.com/questions/44898080/