pytorchltr:在PyTorch中学习排名 源码

PyTorch学习排名(LTR) 这是一个使用PyTorch学习排名(LTR)的库。该库的目标是支持在PyTorch中执行LTR实验所需的基础结构。 安装 在您的virtualenv中,只需运行: pip install pytorchltr 请注意,此库需要Python 3.5或更高版本。 文献资料 文档可。 例子 有关更完整的示例(包括评估),请参见examples/01-basic-usage.py 。 import torch from pytorchltr . datasets import Example3 from pytorchltr . loss import PairwiseHingeLoss # Load dataset train = Example3 ( split = "train" ) collate_fn = train . collate_fn ()