search

yangxudong / deeplearning

深度学习相关的模型训练、评估和预测相关代码
1.01k stars 378 forks source link

请问youtube_match_model.py为何不用原论文的softmax的方式呢？ #23

Open shuDaoNan9 opened 3 years ago

shuDaoNan9 commented 3 years ago

我看YouTubeNet原本的召回阶段使用的是softmax分类，是因为认为这个应该被理解为多分类问题，可以“同时”选择多个“下个”要播放的视频吗？ https://github.com/yangxudong/deeplearning/blob/master/youtube_match_model/youtube_match_model.py loss = tf.nn.sigmoid_cross_entropy_with_logits( labels=labels_one_hot, logits=logits)

谢谢！