基于多头自注意力神经网络的购物篮推荐方法*
倪维健,郭浩宇,刘彤,曾庆田

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