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

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

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

Weijian Ni,Haoyu Guo,Tong Liu,Qingtian Zeng

表4 TaoBao数据集结果

Table 4 Results on TaoBao Dataset

Model Precision Recall F1 Hit-Rate NDCG@5 TOP 0.0011 0.0024 0.0015 0.0048 0.0025 Item-CF 0.0012 0.0034 0.0018 0.0056 0.0022 NFM 0.0074 0.0185 0.0106 0.0307 0.0136 DREAM 0.0020 0.0044 0.0028 0.0088 0.0038 NAM 0.0019 0.0042 0.0026 0.0080 0.0037 ANAM 0.0019 0.0046 0.0027 0.0088 0.0038 本文方法 0.0158 0.0477 0.0237 0.0704 0.0410