type
status
date
slug
summary
tags
category
icon
password
comment
相关链接
pytorch 对应版本下载： https://pytorch.org/get-started/previous-versions/
环境配置
Requirements
- Python 3.8.11
- PyTorch 1.8.0
- numpy 1.23.4
pytorch 安装
在官网上找到的安装 PyTorch 1.8.0 的相关命令
CUDA 10.2 我使用的是这个，试过了，没有问题
代码微调
实际运行时，代码中有几个地方需要调整
src/main.py
src/step_sample.py
复现结果
amazon_beauty
运行时间
2025-11-05_20-28-51
2025-11-05_23-30-26
3小时+32分钟
ml-1m
运行时间
2025-11-11_09-28-56
2025-11-12_02-09-17
4小时+41分钟
steam
运行时间
2025-11-09_21-28-28
2025-11-11_06-58-01
33小时+30分钟
toys
运行时间
2025-11-09_16-24-51
2025-11-09_18-47-35
2个小时+23分钟
总结
Dataset | Metric | DiffuRec |
Beauty | HR@5 | 5.5268 |
ㅤ | HR@10 | 7.792 |
ㅤ | HR@20 | 10.7283 |
ㅤ | NDCG@5 | 3.9762 |
ㅤ | NDCG@10 | 4.7032 |
ㅤ | NDCG@20 | 5.4414 |
Toys | HR@5 | 5.5813 |
ㅤ | HR@10 | 7.3512 |
ㅤ | HR@20 | 9.9543 |
ㅤ | NDCG@5 | 4.1949 |
ㅤ | NDCG@10 | 4.7627 |
ㅤ | NDCG@20 | 5.417 |
Movielens-1M | HR@5 | 15.9027 |
ㅤ | HR@10 | 23.7053 |
ㅤ | HR@20 | 35.6793 |
ㅤ | NDCG@5 | 10.323 |
ㅤ | NDCG@10 | 12.8231 |
ㅤ | NDCG@20 | 15.8409 |
Steam | HR@5 | 6.6808 |
ㅤ | HR@10 | 10.8238 |
ㅤ | HR@20 | 16.775 |
ㅤ | NDCG@5 | 4.2967 |
ㅤ | NDCG@10 | 5.6265 |
ㅤ | NDCG@20 | 7.122 |
- 作者:longlong
- 链接:https://long.long-code.cn//article/diffurec
- 声明:本文采用 CC BY-NC-SA 4.0 许可协议，转载请注明出处。
