-
daisyRec是正在pytorch中开发的一款推荐系统算法
资源介绍
概述
DaisyRec是一个Python工具箱,用于处理评级预测和项目排名问题。
该名DAISY(约:))代表多d imension F A irly COMPAR我的儿子推荐SY干。
要获取所有依赖关系,请运行:
pip install -r requirements.txt
在运行之前,您需要先运行:
python setup.py build_ext --inplace
生成用于进一步导入的.so或.pyd文件。
确保可以加速CUDA环境,因为这些深度学习模型可以以此为基础。我们将不断更新此仓库。
DaisyRec主要处理排名问题,并将推荐问题分为点对问题和成对问题,以构造不同的损失函数,如BPR,Top-1,铰链和交叉熵。下面展示了已经实现的所有算法:
使用main.py可以达到通过上述某些算法计算出的KPI结果。例如,您可以实施此程序来实施BPR-MF:
python m