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

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

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

Weijian Ni,Haoyu Guo,Tong Liu,Qingtian Zeng

表2 Ta-Feng数据集结果

Table 2 Results on Ta-Feng Dataset

Model Precision Recall F1 Hit-Rate NDCG@5 TOP 0.0497 0.0698 0.0580 0.2172 0.0833 Item-CF 0.0272 0.0400 0.0324 0.1244 0.0431 NFM 0.0518 0.0728 0.0605 0.2432 0.0831 DREAM 0.0533 0.0781 0.0633 0.2322 0.0851 NAM 0.0599 0.0941 0.0732 0.2627 0.0925 ANAM 0.0600 0.0942 0.0733 0.2632 0.0925 本文方法 0.0624 0.0968 0.0759 0.2732 0.0940