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

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

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

Weijian Ni,Haoyu Guo,Tong Liu,Qingtian Zeng

表7 TaoBao数据集网络部分性能结果

Table 7 Ablated Results on TaoBao Dataset

Model Precision Recall F1 Hit-Rate NDCG@5 -category-attention 0.0080 0.0221 0.0118 0.0363 0.0178 -category-transformer 0.0046 0.0142 0.0070 0.0209 0.0101 -multihead 0.0171 0.0543 0.0260 0.0738 0.0450 -attention 0.0047 0.0143 0.0071 0.0272 0.0138 -transformer 0.0055 0.0145 0.0080 0.0252 0.0176 完整网络 0.0158 0.0477 0.0237 0.0704 0.0410