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