基于多头自注意力神经网络的购物篮推荐方法*

倪维健,郭浩宇,刘彤,曾庆田

Online Product Recommendation Based on Multi-Head Self-Attention Neural Networks

Weijian Ni,Haoyu Guo,Tong Liu,Qingtian Zeng

表3 JingDong数据集结果

Table 3 Results on JingDong Dataset

Model Precision Recall F1 Hit-Rate NDCG@5 TOP 0.0140 0.0392 0.0206 0.0679 0.0270 Item-CF 0.0123 0.0341 0.0181 0.0584 0.0251 NFM 0.0337 0.0822 0.0478 0.1409 0.0633 DREAM 0.0213 0.0592 0.0314 0.0988 0.0413 NAM 0.0257 0.0678 0.0372 0.1150 0.0485 ANAM 0.0568 0.1530 0.0828 0.2294 0.1226 本文方法 0.0678 0.1859 0.0994 0.2620 0.1485