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

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

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

Weijian Ni,Haoyu Guo,Tong Liu,Qingtian Zeng

表6 JingDong数据集网络部分性能结果

Table 6 Ablated Results on JingDong Dataset

Model Precision Recall F1 Hit-Rate NDCG@5 -category-attention 0.0428 0.1157 0.0625 0.1835 0.0919 -category-transformer 0.0559 0.1506 0.0816 0.2225 0.1232 -multihead 0.0665 0.1781 0.0969 0.2577 0.1480 -attention 0.0428 0.1223 0.0634 0.1820 0.0930 -transformer 0.0568 0.1539 0.0829 0.2304 0.1256 完整网络 0.0678 0.1859 0.0994 0.2620 0.1485