相关项目
allRank
allRank是一个基于PyTorch的框架，旨在简化神经排序学习模型的实验。它提供多种损失函数和评分函数，并支持常用评估指标如NDCG和MRR。该框架支持添加自定义损失和配置模型与训练流程，适用于研究和工业应用。同时支持GPU和CPU架构，并集成了Google云存储功能。
ranking
TensorFlow Ranking是一个适用于学习排名（LTR）技术的开源库，基于TensorFlow平台。该库包括点对、成对和列表损失函数，以及各类排名指标如平均倒数排名（MRR）和标准折扣累积增益（NDCG）。它提供了群组评分功能和LambdaLoss的实现，用于优化排名指标。此外，它还支持从偏见反馈数据中进行无偏学习。该库旨在为学术研究和工业应用提供一个开放、便捷的平台。用户可通过教程和演示快速入门，无需复杂安装。