代码逻辑整理

本文详细介绍了构建神经网络的过程,包括输入部分的用户和电影相关数据,通过嵌入层和全连接层处理用户特征,电影ID的嵌入,以及电影类型和标题的卷积网络。还涉及了Dropout层和最终的全连接层,提供了AFM模型的源码实现。
摘要由CSDN通过智能技术生成

构建神经网络

  1. 输入
    函数 get_inputs()
    user相关:uid, user_gender,user_age,user_job
    movie相关:movie_id,movie_categories,movie_titles

  2. 定义user矩阵
    函数 get_user_embedding( )
    uid_embed_layer,gender_embed_layer,age_embed_layer,job_embed_layer

  3. 将User的嵌入矩阵一起全连接生成User的特征
    函数 get_user_feature_layer( )
    第一层全连接. uid_fc_layer,gender_fc_layer, age_fc_layer, job_fc_layer
    第二层全连接 user_combine_layer,user_combine_layer,user_combine_layer_flat

  4. 定义Movie ID的嵌入矩阵
    函数 get_movie_id_embed_layer()
    movie_id_embed_layer

  5. 合并电影类型的多个嵌入向量
    函数 get_movie_categories_layers()
    movie_categories_embed_layer, movie_categories_embed_layer

  6. Movie Title的文本卷积网络实现
    函数 get_movie_cnn_layer()
    从嵌入矩阵中得到电影名对应的各个单词的嵌入向量
    movi

  • 0
    点赞
  • 1
    收藏
    觉得还不错? 一键收藏
  • 0
    评论
评论
添加红包

请填写红包祝福语或标题

红包个数最小为10个

红包金额最低5元

当前余额3.43前往充值 >
需支付:10.00
成就一亿技术人!
领取后你会自动成为博主和红包主的粉丝 规则
hope_wisdom
发出的红包
实付
使用余额支付
点击重新获取
扫码支付
钱包余额 0

抵扣说明:

1.余额是钱包充值的虚拟货币,按照1:1的比例进行支付金额的抵扣。
2.余额无法直接购买下载,可以购买VIP、付费专栏及课程。

余额充值