Beibei, Tmall, IJCAI
收藏资源简介:
我们使用三个数据集来评估MixRec:Beibei、Tmall和IJCAI。我们采用两个代表性指标来评估top-N物品推荐的准确性:命中率(HR@N)和归一化折现累积增益(NDCG@N)。所有负样本都用于构建测试集,其中包含用户在目标行为类型下的所有正向交互。
We utilized three datasets to evaluate MixRec: Beibei, Tmall, and IJCAI. We employed two representative metrics to assess the accuracy of top-N item recommendation: Hit Rate (HR@N) and Normalized Discounted Cumulative Gain (NDCG@N). All negative samples are used to construct the test set, which contains all positive interactions of users under the target behavior type.
MixRec: 异构图协同过滤
数据集
MixRec使用了三个数据集进行评估:Beibei、Tmall和IJCAI。评估指标包括Hit Ratio (HR@N)和Normalized Discounted Cumulative Gain (NDCG@N)。采用leave-one-out评估策略,所有负样本用于构建测试集,包含用户的所有正向交互。
| 数据集 | 用户数 | 物品数 | 交互数 | 交互密度 |
|---|---|---|---|---|
| Beibei | 21716 | 7977 | 282860 | 0.1633% |
| Tmall | 114503 | 66706 | 491870 | 0.0064% |
| IJCAI | 423423 | 874328 | 2926616 | 0.0008% |
使用方法
需要先解压Tmall和IJCAI数据集,并创建History/和Models/目录。切换工作目录到MixRec/。训练命令如下:
- Beibei
python mixrec_bei.py --data beibei --reg 1 --batch 32
- Tmall
python mixrec.py --data tmall --ssl_reg 1e-6 --reg 5e-5 --keepRate 0.4 --graphSampleN 20000 --testgraphSampleN 40000
- IJCAI
python mixrec.py --data ijcai --lr 1e-4 --graphSampleN 20000 --testgraphSampleN 40000
重要参数
reg:权重衰减正则化的权重,从集合{1e-2, 1e-3, 1e-4, 1e-5}中调整。ssl_reg和sslGlobal_reg:节点级和图级对比目标的权重衰减正则化的权重,从集合{1e-4, 1e-5, 1e-6, 1e-7}中调整。graphSampleN:训练期间子图节点数,推荐值为{10000, 15000, 20000, 25000, 30000}。testgraphSampleN:测试期间子图节点数,推荐值为{30000, 35000, 40000, 45000, 50000}。




