search

shenweichen / DeepMatch

A deep matching model library for recommendations & advertising. It's easy to train models and to export representation vectors which can be used for ANN search.
https://deepmatch.readthedocs.io/en/latest/
Apache License 2.0
2.22k stars 530 forks source link

YoutubeDNN是不是少了个隐含层？ #47

Closed shuDaoNan9 closed 3 years ago

shuDaoNan9 commented 3 years ago

原论文，整个模型架构是包含三个隐层的DNN结构。但是我看这里user_dnn_hidden_units只有两层？ def YoutubeDNN(user_feature_columns, item_feature_columns, num_sampled=5, user_dnn_hidden_units=(64, 32), dnn_activation='relu', dnn_use_bn=False, l2_reg_dnn=0, l2_reg_embedding=1e-6, dnn_dropout=0, output_activation='linear', seed=1024, ): 有小伙伴知道原因否？谢谢！