ReChorus:推荐模型的“合唱”:具有隐式反馈的Top K推荐的PyTorch框架 源码

ReChorus是用于Top-K建议的通用PyTorch框架,具有隐式反馈,尤其是用于研究目的。 它旨在提供一个公平的基准,以比较不同的最新算法。 我们希望这可以部分缓解不同论文采用无法比拟的实验设置的问题,从而形成推荐算法的“合唱”。 该框架特别适合研究人员比较相同实验设置下的算法,以及新手熟悉经典方法的情况。 我们框架的特征可以总结如下: 敏捷:在一个文件中专注于模型设计并快速实施新模型 容易:该框架用不到一千行代码完成,易于使用干净的代码和足够的注释 高效:多线程批处理准备,用于评估的特殊实现以及在深度模型训练期间约90%的GPU利用率 灵活:针对不同的数据集和实验设置实施新的阅读器