Neural Collaborative Filtering复现

环境

tensorflow-gpu:2.4
keras:2.3.1

在原代码上需进行的修改:
1.NeuMF.py get_model函数

    # Embedding layer
    MF_Embedding_User = Embedding(input_dim = num_users, output_dim = mf_dim,
                                  name = 'mf_embedding_user',input_length=1)
    MF_Embedding_Item = Embedding(<

你可能感兴趣的:(推荐论文,python,深度学习,tensorflow,keras)