Neural network based Collaborative Filtering

NCF

1. 论文

Neural network-based Collaborative Filtering

创新：使用深度学习的方法进行协同过滤

2. 模型结构

3. 实验数据集

采用Movielens数据集进行测试。数据集的处理见NCF文件。

4. 模型API

class NeuMF(keras.Model):
    def __init__(self, num_users, num_items, mf_dim, layers, reg_layers, reg_mf):
        super(NeuMF, self).__init__()
        self.MF_Embedding_User = keras.layers.Embedding(
            input_dim=num_users,
            output_dim=mf_dim,
            name='mf_embedding_user',
            embeddings_initializer='random_uniform',
            embeddings_regularizer=regularizers.l2(reg_mf[0]),
        )