文件名称:ptranking:在PyTorch中学习排名
文件大小:1.04MB
文件格式:ZIP
更新时间:2024-05-30 06:18:42
pytorch ranking learning-to-rank Python
介绍 这个开源项目称为PTRanking (在PyTorch中学习排名),旨在提供基于PyTorch的典型的按等级学习方法的可扩展和可扩展的实现。 一方面,该项目可以对多个基准数据集进行统一比较,从而可以深入了解以前的学习排名方法。 另一方面,该项目使开发和合并新提出的模型变得容易,从而扩展了等级学习技术的领域。 主要特点: 许多具有代表性的学习排名模型,不仅包括通过经验风险最小化的传统优化框架,还包括对抗性优化框架 支持广泛使用的基准数据集。 同时,还支持以指定比例随机掩盖真实标签 支持不同的指标,例如Precision,MAP,nDCG和nERR 高度可配置的功能,用于微调超参数,例如,对特定模型的超参数进行网格搜索 提供易于使用的API,以开发新的按等级学习模型 请参考以获取更多详细信息。