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

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

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

Weijian Ni,Haoyu Guo,Tong Liu,Qingtian Zeng

表5 Ta-Feng数据集网络部分性能结果

Table 5 Ablated Results on Ta-Feng Dataset

Model Precision Recall F1 Hit-Rate NDCG@5 -category-attention 0.0589 0.0935 0.0723 0.2629 0.0930 -category-transformer 0.0533 0.0746 0.0621 0.2317 0.0686 -multihead 0.0617 0.0961 0.0752 0.2687 0.0934 -attention 0.0601 0.0943 0.0734 0.2634 0.0787 -transformer 0.0556 0.0899 0.0687 0.2512 0.0836 完整网络 0.0624 0.0968 0.0759 0.2732 0.0940