huangjunheng / recommendation_model

练习下用pytorch来复现下经典的推荐系统模型, 如MF, FM, DeepConn, MMOE, PLE, DeepFM, NFM, DCN, AFM, AutoInt, ONN, FiBiNET, DCN-v2, AFN, DCAP等
505 stars 116 forks source link

有个bug #4

Open ZJUguquan opened 2 years ago

ZJUguquan commented 2 years ago

dcn_v2/dcn_v2.py 第47行,应该是 fc = self.dropout(fc) ,减号应该换成等号~

huangjunheng commented 2 years ago

谢谢 已修改